Update app.py

app.py

@@ -2,10 +2,18 @@
 import warnings
 warnings.filterwarnings("ignore")
 
+import subprocess
 import gradio as gr
 from src.florence.task import ocr_task
 
 
+# Install Flash Attention
+subprocess.run(
+    "pip install flash-attn --no-build-isolation",
+    env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
+    shell=True,
+)
+
 # Image input for the interface
 image = gr.Image(type="pil", label="Image")