Update app.py

app.py

@@ -1,7 +1,16 @@
+import os
 import gradio as gr
 from huggingface_hub import InferenceClient
 from transformers import AutoModelForCausalLM, pipeline
 
+# Ensure necessary packages are installed
+def install(package):
+    os.system(f"pip install {package}")
+
+# Install torch and flash_attn
+install("torch")
+install("flash_attn")
+
 # Use a pipeline as a high-level helper
 pipe = pipeline("visual-question-answering", model="dandelin/vilt-b32-finetuned-vqa", trust_remote_code=True)