Ajustar función responder para modelo tiny-gpt2

app.py

@@ -4,9 +4,10 @@ from transformers import pipeline
 generator = pipeline("text-generation", model="sshleifer/tiny-gpt2")
 
 def responder(texto_usuario):
-        prompt = f"Tú eres AURA, acrónimo de Asistente de Unión y Regulación Afectiva, un asistente emocional escolar muy amable de la aplicación CEJASOS. Ayuda a los estudiantes a sentirse escuchados sin juzgar y de una manera en la que se sientan seguros contigo. Si alguien dice: '{texto_usuario}', tú respondes:"
-    resultado = generator(prompt, max_length=150, do_sample=True, temperature=0.7)[0]["text"]
-    return resultado[len(prompt):]
+    prompt = f"Tú eres AURA, un asistente emocional escolar muy amable. Ayuda a los estudiantes a sentirse escuchados sin juzgar. Si alguien dice: '{texto_usuario}', tú respondes:"
+    outputs = generator(prompt, max_length=50)
+    return outputs[0]["text"]
+
 
 iface = gr.Interface(
     fn=responder,