Merlintxu commited on
Commit
c95135b
·
1 Parent(s): 9a2d9be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import spacy
3
  from transformers import pipeline
4
 
5
  nlp = spacy.load('es_core_news_sm')
6
- text_generator = pipeline('text-generation', model='microsoft/DialoGPT-large')
7
 
8
  pos_tags = ['ADJ', 'ADP', 'ADV', 'AUX', 'CONJ', 'DET', 'INTJ', 'NOUN', 'NUM', 'PART', 'PRON', 'PROPN', 'PUNCT', 'SCONJ', 'SYM', 'VERB', 'X']
9
 
 
3
  from transformers import pipeline
4
 
5
  nlp = spacy.load('es_core_news_sm')
6
+ text_generator = pipeline('text-generation', model='datificate/gpt-2-small-spanish')
7
 
8
  pos_tags = ['ADJ', 'ADP', 'ADV', 'AUX', 'CONJ', 'DET', 'INTJ', 'NOUN', 'NUM', 'PART', 'PRON', 'PROPN', 'PUNCT', 'SCONJ', 'SYM', 'VERB', 'X']
9