Spaces:
Runtime error
Runtime error
| 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() | |