Update app.py

app.py

@@ -2,12 +2,15 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 import torch
 import requests
+import subprocess
 
-# Ensure torch is installed before flash_attn
+# Ensure flash-attn is installed before using it
 try:
     import flash_attn
 except ImportError:
-    raise RuntimeError("Missing required dependency: flash_attn. Install with `pip install flash-attn --no-build-isolation`")
+    print("Installing flash-attn...")
+    subprocess.run(["pip", "install", "flash-attn"], check=True)
+    import flash_attn
 
 # Load DeepSeek-R1 model
 model_name = "deepseek-ai/DeepSeek-R1"
@@ -77,4 +80,3 @@ demo = gr.Interface(
 if __name__ == "__main__":
     demo.launch()
 
-