Update app.py

app.py

@@ -1,10 +1,30 @@
-# Example: Gradio app for Zephyr-7B
 import gradio as gr
 from transformers import pipeline
+import time
 
-model = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta")
+# Initialize model (replace with your actual setup)
+model = pipeline("text-generation", model="gpt2")
 
-def chat(prompt):
-    return model(prompt, max_new_tokens=100)[0]['generated_text']
+def respond(prompt):
+    if not prompt.strip():
+        return "Please enter a message"
+    
+    print(f"Processing: {prompt}")  # Debug log
+    start_time = time.time()
+    
+    try:
+        response = model(prompt, max_length=50)[0]['generated_text']
+        print(f"Response: {response}")  # Debug log
+        return response
+    except Exception as e:
+        print(f"Error: {str(e)}")
+        return f"Error: {str(e)}"
 
-gr.Interface(fn=chat, inputs="textbox", outputs="textbox").launch()
+# Launch interface
+gr.Interface(
+    fn=respond,
+    inputs="textbox",
+    outputs="textbox",
+    title="My AI Assistant",
+    description="Type something and press Enter"
+).launch()