Limpiar respuesta para evitar repetir prompt
Browse filesHacer que la respuesta de AURA esté limpia y no brinde la información inicial más la respuesta, sino solamente la respuesta
app.py
CHANGED
@@ -5,8 +5,9 @@ generator = pipeline("text-generation", model="sshleifer/tiny-gpt2")
|
|
5 |
|
6 |
def responder(texto_usuario):
|
7 |
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:"
|
8 |
-
|
9 |
-
|
|
|
10 |
|
11 |
|
12 |
iface = gr.Interface(
|
|
|
5 |
|
6 |
def responder(texto_usuario):
|
7 |
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:"
|
8 |
+
resultado_completo = generator(prompt, max_length=50)[0]["text"]
|
9 |
+
respuesta_limpia = resultado_completo[len(prompt):].strip()
|
10 |
+
return respuesta_limpia
|
11 |
|
12 |
|
13 |
iface = gr.Interface(
|