Intelarya.ai / app.py
hexanovapixel's picture
Update app.py
6c06335 verified
import gradio as gr
from transformers import pipeline
# Modelo
generator = pipeline("text-generation", model="gpt2")
def responder(message, history):
history = history or []
prompt = "Eres Intelarya.ai, una IA educativa que explica paso a paso de forma clara.\n\n"
for msg in history:
if msg["role"] == "user":
prompt += f"Usuario: {msg['content']}\n"
else:
prompt += f"Intelarya: {msg['content']}\n"
prompt += f"Usuario: {message}\nIntelarya:"
result = generator(prompt, max_length=150, do_sample=True, temperature=0.7)
response = result[0]["generated_text"].split("Intelarya:")[-1].strip()
history.append({"role": "user", "content": message})
history.append({"role": "assistant", "content": response})
return history, history
with gr.Blocks() as demo:
gr.Markdown("# 💎 intelarya.ai")
gr.Markdown("Aprende más rápido. Entiende mejor. Sin complicaciones.")
chatbot = gr.Chatbot(type="messages")
msg = gr.Textbox(placeholder="Escribe tu tarea aquí...")
state = gr.State([])
msg.submit(responder, [msg, state], [chatbot, state])
demo.launch()