wjbmattingly commited on
Commit
d2e31b4
1 Parent(s): 483be0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -15,6 +15,7 @@ import os
15
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
16
 
17
  model_dir = 'medieval-data/florence2-medieval-bbox-line-detection'
 
18
 
19
  def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
20
  """Work around for https://huggingface.co/microsoft/phi-1_5/discussions/72."""
 
15
  subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
16
 
17
  model_dir = 'medieval-data/florence2-medieval-bbox-line-detection'
18
+ model_dir = "medieval-data/florence2-medieval-bbox-zone-detection"
19
 
20
  def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
21
  """Work around for https://huggingface.co/microsoft/phi-1_5/discussions/72."""