Spaces:
Sleeping
Sleeping
wjbmattingly
commited on
Commit
•
d2e31b4
1
Parent(s):
483be0d
Update app.py
Browse files
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."""
|