Update app.py

app.py

@@ -1,12 +1,19 @@
-import gradio as gr
 from transformers import pipeline
+import gradio as gr
 
-# Modelo ligero para bots rápidos
-generator = pipeline("text-generation", model="microsoft/DialoGPT-medium")
+chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1", trust_remote_code=True)
 
-def responder(mensaje):
-    respuesta = generator(mensaje, max_new_tokens=100, do_sample=True, top_p=0.9)[0]['generated_text']
-    return respuesta.split(mensaje)[-1].strip()
+def predict(user_input, history=[]):
+    prompt = user_input
+    response = chatbot(prompt, max_new_tokens=200, do_sample=True)[0]["generated_text"]
+    history.append((user_input, response))
+    return history, history
 
-iface = gr.Interface(fn=responder, inputs="text", outputs="text", title="Bot Coqueto", theme="soft")
-iface.launch()
+gr.Interface(
+    fn=predict,
+    inputs=["text", "state"],
+    outputs=["chatbot", "state"],
+    title="Mi Chatbot Sexy",
+    description="Habla con una IA encantadora.",
+    theme="finlaymacklon/boxy_violet"
+).launch()