Update app.py

app.py

@@ -1,12 +1,14 @@
-from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
+import subprocess
+import sys
+
+subprocess.check_call([sys.executable, "-m", "pip", "install", "transformers", "torch", "gradio"])
+
 import gradio as gr
+from transformers import pipeline
 
 pipe = pipeline("text-generation", model="sapientinc/HRM-Text-1B")
-tokenizer = AutoTokenizer.from_pretrained("sapientinc/HRM-Text-1B")
-model = AutoModelForCausalLM.from_pretrained("sapientinc/HRM-Text-1B")
 
 def chat(message, history):
-    result = pipe(message, max_length=200)
-    return result[0]["generated_text"]
+    return pipe(message, max_length=200)[0]["generated_text"]
 
 gr.ChatInterface(fn=chat).launch()