File size: 957 Bytes
3926df8 3832021 3926df8 ba7c9a5 3926df8 70d4b9f 3832021 70d4b9f c3b01dd 3926df8 3832021 3926df8 3832021 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
import gradio as gr
from transformers import pipeline
generator = pipeline("text-generation", model="sshleifer/tiny-gpt2")
# Variable para almacenar el historial de conversación
historial = ""
def responder(texto_usuario):
global historial
prompt_inicial = "Tú eres AURA, un asistente emocional escolar muy amable. Escucha sin juzgar y responde con empatía."
# Construir el nuevo prompt con el historial
prompt = (
f"{prompt_inicial}\n"
f"{historial}"
f"Usuario: {texto_usuario}\n"
f"AURA:"
)
resultado_completo = generator(prompt, max_length=100)[0]["text"]
respuesta = resultado_completo[len(prompt):].strip()
# Actualizar el historial
historial += f"Usuario: {texto_usuario}\nAURA: {respuesta}\n"
return respuesta
iface = gr.Interface(
fn=responder,
inputs="text",
outputs="text",
title="AURA",
description="Tu espacio seguro 💜"
)
iface.launch()
|