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

Usar tiny-gpt2 para evitar errores

Browse files

Switched to a lighter model to fix loading error

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- generator = pipeline("text-generation", model="EleutherAI/gpt-neo-1.3B")
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, la cual es para proteger la integridad de los alumnos en situaciones de emergencia. 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
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ 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