Update app.py

app.py

@@ -9,10 +9,20 @@ import numpy as np
 import requests
 from io import BytesIO
 from unittest.mock import patch
+from transformers.dynamic_module_utils import get_imports
 
 subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
 
 model_dir = 'medieval-data/florence2-medieval-bbox-line-detection'
+
+def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
+    """Work around for https://huggingface.co/microsoft/phi-1_5/discussions/72."""
+    if not str(filename).endswith("/modeling_florence2.py"):
+        return get_imports(filename)
+    imports = get_imports(filename)
+    imports.remove("flash_attn")
+    return imports
+    
 with patch("transformers.dynamic_module_utils.get_imports", fixed_get_imports):
     # Load the configuration
     config = AutoConfig.from_pretrained(model_dir, trust_remote_code=True)