Update app.py

app.py

@@ -13,7 +13,7 @@ from unittest.mock import patch
 from transformers.dynamic_module_utils import get_imports
 
 def fixed_get_imports(filename: str | os.PathLike) -> list[str]:
-    if not str(filename).endswith("flash_attention.py"):
+    if not str(filename).endswith("modeling_sa2va_chat.py"):
         return get_imports(filename)
     imports = get_imports(filename)
     imports.remove("flash_attn")