GrupoCEJA commited on
Commit
22183a4
verified
1 Parent(s): ba7c9a5

Corregir clave de texto para el modelo tiny-gpt2

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ generator = pipeline("text-generation", model="sshleifer/tiny-gpt2")
5
 
6
  def responder(texto_usuario):
7
  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:"
8
- resultado = generator(prompt, max_length=150, do_sample=True, temperature=0.7)[0]["generated_text"]
9
  return resultado[len(prompt):]
10
 
11
  iface = gr.Interface(
 
5
 
6
  def responder(texto_usuario):
7
  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:"
8
+ resultado = generator(prompt, max_length=150, do_sample=True, temperature=0.7)[0]["text"]
9
  return resultado[len(prompt):]
10
 
11
  iface = gr.Interface(