Update app.py

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 from transformers import pipeline
 import requests
 
-# Use a model that supports conversational tasks
-standard_model = pipeline("conversational", model="google/mt5-base")
-premium_model = pipeline("conversational", model="google/mt5-base")
+# Use the text-generation task
+standard_model = pipeline("text-generation", model="google/mt5-base")
+premium_model = pipeline("text-generation", model="google/mt5-base")
 
 # Your Boosty API key or any method of checking subscriptions
 BOOSTY_API_KEY = "YOUR_BOOSTY_API_KEY"
@@ -30,7 +30,7 @@ def chat_with_ai(user_input, boosty_user_id):
 
     # Get AI response
     conversation = model_to_use(user_input)
-    return conversation[-1]["generated_text"]
+    return conversation[0]["generated_text"]
 
 # Create the Gradio interface
 iface = gr.Interface(