update

app.py

@@ -20,6 +20,9 @@ import torch
 from transformers import AutoModelForCausalLM
 from transformers import AutoProcessor 
 
+# add a command for installing flash-attn
+os.system('pip install flash-attn --no-build-isolation')
+
 # Define repository and local directory
 repo_id = "microsoft/OmniParser-v2.0"  # HF repo
 local_dir = "weights"  # Target local directory

requirements.txt

@@ -37,5 +37,4 @@ easyocr
 paddleocr
 paddlepaddle==2.6.2
 supervision==0.18.0
-ultralytics==8.3.78
-flash-attn
+ultralytics==8.3.78