Update app.py

app.py

@@ -1,18 +1,20 @@
 from transformers import pipeline
 import gradio as gr
 
-# ✅ Use Falcon-7B-Instruct now that you're on GPU
-generator = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", device_map="auto")
+# Upgrade to FLAN-UL2 — more detailed, better reasoning
+generator = pipeline("text2text-generation", model="google/flan-ul2")
 
 def interpret_prompt(prompt):
     system_prompt = (
-        f"A user has typed this prompt to an AI: \"{prompt}\"\n"
-        "List 3 different ways the AI could interpret this prompt, clearly labeled.\n"
-        "Then give 1 tip for how the user could improve their prompt to be more specific or useful."
+        f"A user gave this prompt to an AI: \"{prompt}\"\n"
+        "What are three different ways the AI could interpret this prompt?\n"
+        "Number each one.\n"
+        "Then give one short tip for how the user could improve their prompt."
     )
 
-    output = generator(system_prompt, max_new_tokens=300, do_sample=True, temperature=0.7)[0]['generated_text']
+    output = generator(system_prompt, max_length=400)[0]['generated_text']
 
+    # Optional formatting fix
     if "Tip" in output:
         parts = output.split("Tip", 1)
         interpretations = parts[0].strip()