Update app.py

app.py

@@ -1,45 +1,45 @@
-import gradio as gr
-from transformers import pipeline
-import requests
-
-# Load both the standard and premium models
-standard_model = pipeline("conversational", model="microsoft/DialoGPT-medium")
-premium_model = pipeline("conversational", model="microsoft/DialoGPT-large")
-
-# Your Boosty API key or any method of checking subscriptions
-BOOSTY_API_KEY = "YOUR_BOOSTY_API_KEY"
-BOOSTY_API_URL = "https://boosty.to/api/subscription_check"  # Replace with the actual Boosty API URL
-
-# Function to check if a user is a premium member on Boosty
-def is_premium_user(boosty_user_id: str) -> bool:
-    response = requests.get(f"{BOOSTY_API_URL}?user_id={boosty_user_id}", headers={"Authorization": f"Bearer {BOOSTY_API_KEY}"})
-    
-    if response.status_code != 200:
-        return False  # Assuming non-200 response means not premium
-    
-    # Assuming the response contains a field like 'is_premium': True/False
-    return response.json().get("is_premium", False)
-
-# Chat function to use the correct model based on Boosty subscription
-def chat_with_ai(user_input, boosty_user_id):
-    # Check if the user is a premium member
-    if is_premium_user(boosty_user_id):
-        model_to_use = premium_model
-    else:
-        model_to_use = standard_model
-
-    # Get AI response
-    conversation = model_to_use(user_input)
-    return conversation[-1]["generated_text"]
-
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=chat_with_ai, 
-    inputs=["text", gr.Textbox(label="Boosty User ID")], 
-    outputs="text", 
-    live=True, 
-    title="AI Chatbot"
-)
-
-# Launch the Gradio app as an API
-iface.launch(server_name="0.0.0.0", server_port=7860, share=True)
+import gradio as gr
+from transformers import pipeline
+import requests
+
+# Load both the standard and premium models
+standard_model = pipeline("conversational", model="google/mt5-base")
+premium_model = pipeline("conversational", model="google/mt5-base")
+
+# Your Boosty API key or any method of checking subscriptions
+BOOSTY_API_KEY = "YOUR_BOOSTY_API_KEY"
+BOOSTY_API_URL = "https://boosty.to/api/subscription_check"  # Replace with the actual Boosty API URL
+
+# Function to check if a user is a premium member on Boosty
+def is_premium_user(boosty_user_id: str) -> bool:
+    response = requests.get(f"{BOOSTY_API_URL}?user_id={boosty_user_id}", headers={"Authorization": f"Bearer {BOOSTY_API_KEY}"})
+    
+    if response.status_code != 200:
+        return False  # Assuming non-200 response means not premium
+    
+    # Assuming the response contains a field like 'is_premium': True/False
+    return response.json().get("is_premium", False)
+
+# Chat function to use the correct model based on Boosty subscription
+def chat_with_ai(user_input, boosty_user_id):
+    # Check if the user is a premium member
+    if is_premium_user(boosty_user_id):
+        model_to_use = premium_model
+    else:
+        model_to_use = standard_model
+
+    # Get AI response
+    conversation = model_to_use(user_input)
+    return conversation[-1]["generated_text"]
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=chat_with_ai, 
+    inputs=["text", gr.Textbox(label="Boosty User ID")], 
+    outputs="text", 
+    live=True, 
+    title="AI Chatbot"
+)
+
+# Launch the Gradio app as an API
+iface.launch(server_name="0.0.0.0", server_port=7860, share=True)