SebastianItamari commited on
Commit
bc0bf41
·
verified ·
1 Parent(s): a3ee688

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  # Inicializar el modelo generador de texto
5
  @st.cache_resource # Esto cachea el modelo para no cargarlo repetidamente
6
  def load_model():
7
- return pipeline('text-generation', model='distilgpt2') # Cambia el modelo a distilgpt2
8
 
9
  generator = load_model()
10
 
 
4
  # Inicializar el modelo generador de texto
5
  @st.cache_resource # Esto cachea el modelo para no cargarlo repetidamente
6
  def load_model():
7
+ return pipeline('text-generation', model='gpt2') # Cambia el modelo a distilgpt2
8
 
9
  generator = load_model()
10