Update app.py

app.py

@@ -14,7 +14,14 @@ import uuid
 import uuid
 from ultralytics import YOLO
 import gradio as gr
-
+import subprocess
+import sys
+try:
+    # Run the pip command
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "flash-attn", "--no-build-isolation"])
+    print("flash-attn installed successfully.")
+except subprocess.CalledProcessError as e:
+    print(f"An error occurred while installing flash-attn: {e}")
 # # default: Load the model on the available device(s)
 # model = Qwen2VLForConditionalGeneration.from_pretrained(
 #     "Qwen/Qwen2-VL-7B-Instruct", torch_dtype="auto", device_map="auto"