Update app.py

app.py

@@ -1,16 +1,28 @@
 import os
-import gradio as gr
-from huggingface_hub import InferenceClient
-from transformers import AutoModelForCausalLM, pipeline
+import subprocess
+import sys
 
-# Ensure necessary packages are installed
 def install(package):
-    os.system(f"pip install {package}")
+    subprocess.check_call([sys.executable, "-m", "pip", "install", package])
 
-# Install torch and flash_attn
+# Install torch first
 install("torch")
+
+# Install flash_attn next
 install("flash_attn")
 
+# Now install other dependencies
+install("huggingface_hub==0.22.2")
+install("transformers")
+install("openai")
+install("gradio")
+install("einops")
+install("timm")
+
+import gradio as gr
+from huggingface_hub import InferenceClient
+from transformers import AutoModelForCausalLM, pipeline
+
 # Use a pipeline as a high-level helper
 pipe = pipeline("visual-question-answering", model="dandelin/vilt-b32-finetuned-vqa", trust_remote_code=True)