Spaces:
Running
Running
import gradio as gr | |
from transformers import pipeline | |
# Cargar un modelo de lenguaje | |
chatbot = pipeline("text-generation", model="gpt2") | |
# Función de respuesta | |
def responder_entrada(user_input): | |
respuesta = chatbot(user_input, max_length=100, num_return_sequences=1) | |
return respuesta[0]["generated_text"] | |
# Interfaz Gradio | |
with gr.Blocks() as interfaz: | |
gr.Markdown("# Asistente Virtual") | |
chat = gr.Chatbot() | |
entrada_usuario = gr.Textbox(label="Escribe tu mensaje:") | |
boton_enviar = gr.Button("Enviar") | |
def enviar_mensaje(user_input, history): | |
history = history or [] | |
respuesta = responder_entrada(user_input) | |
history.append((user_input, respuesta)) | |
return history, "" | |
boton_enviar.click(enviar_mensaje, [entrada_usuario, chat], [chat, entrada_usuario]) | |
interfaz.launch() |