Update app.py

app.py

@@ -2,23 +2,27 @@
 import gradio as gr
 from transformers import pipeline
 
-# Small multilingual model (Nepali support)
+# Load multilingual T5 (supports Nepali)
 chatbot = pipeline(
     "text2text-generation",
-    model="facebook/xglm-564M",
-    max_new_tokens=128
+    model="google/mt5-base",
+    max_new_tokens=128,
+    temperature=0.7
 )
 
 def chat_nepali(message, history):
+    # Concatenate history into a single prompt
     history_text = ""
     for user_msg, bot_msg in history:
         history_text += f"User: {user_msg}\nBot: {bot_msg}\n"
     prompt = f"{history_text}User: {message}\nBot:"
+    
+    # Generate response
     result = chatbot(prompt)[0]["generated_text"]
     return result.strip()
 
 with gr.Blocks() as demo:
-    gr.Markdown("## Nepali Chatbot 🇳🇵 (Lightweight for Free Tier)")
+    gr.Markdown("## Nepali Chatbot 🇳🇵 using mT5")
     chatbot_ui = gr.Chatbot()
     msg = gr.Textbox(placeholder="नेपालीमा कुरा गर्नुहोस्...")
     clear = gr.Button("Clear Chat")
@@ -27,4 +31,4 @@ with gr.Blocks() as demo:
     clear.click(lambda: None, None, chatbot_ui, queue=False)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()