Update app.py

app.py

@@ -1,37 +1,33 @@
 import gradio as gr
 from transformers import pipeline
 
-# ✅ Load FLAN-T5 model for polite, instruction-based replies
-generator = pipeline("text2text-generation", model="google/flan-t5-small")
+# 🚀 Use lightweight, faster FLAN model
+generator = pipeline("text2text-generation", model="google/flan-t5-xs")
 
-# 📌 Instruction template
+# ✨ Prompt Template
 TEMPLATE = (
-    "You are a helpful, polite, and professional customer support agent. "
-    "Reply to this customer message in a brand-consistent tone:\n\n"
-    "{input}"
+    "You are a polite and professional customer support agent. "
+    "Please respond to the customer's message:\n\n{input}"
 )
 
-# 🔁 Reply Generator
+# 🎯 Response Function
 def generate_reply(user_input):
     prompt = TEMPLATE.format(input=user_input)
-    response = generator(prompt, max_length=150, do_sample=False)[0]["generated_text"]
+    response = generator(prompt, max_length=80, do_sample=False)[0]["generated_text"]
     return response.strip()
 
-# 🎛️ Gradio Interface
+# 🎛 Interface
 iface = gr.Interface(
     fn=generate_reply,
     inputs=gr.Textbox(lines=6, label="Customer Message", placeholder="Enter complaint or question..."),
-    outputs=gr.Textbox(label="Auto-Generated Support Reply"),
-    title="🤖 Auto-Reply Generator for Customer Support",
-    description=(
-        "Generate fast, polite, and professional replies to customer queries using Google's FLAN-T5 model. "
-        "Perfect for CRM bots, helpdesk automation, and ticket response."
-    ),
+    outputs=gr.Textbox(label="Support Reply"),
+    title="⚡ Fast Auto-Reply Generator for Customer Support",
+    description="Fast & polite auto-replies to customer complaints using a tiny FLAN-T5 model. Perfect for CRM or helpdesk automation.",
     examples=[
         ["I still haven't received my order and it's been 10 days."],
-        ["My refund hasn't been processed yet."],
-        ["Your app keeps crashing on my iPhone."],
-        ["Great service, just wanted to say thanks!"]
+        ["Why was I charged twice for my subscription?"],
+        ["Thanks for the quick response yesterday!"],
+        ["My login isn't working since the update."]
     ]
 )