GrupoCEJA commited on
Commit
3c4a92f
·
verified ·
1 Parent(s): c3b01dd

Limpiar respuesta para evitar repetir prompt

Browse files

Hacer que la respuesta de AURA esté limpia y no brinde la información inicial más la respuesta, sino solamente la respuesta

Files changed (1) hide show
  1. app.py +3 -2
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
- outputs = generator(prompt, max_length=50)
9
- return outputs[0]["text"]
 
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(