azaninello commited on
Commit
61395f5
1 Parent(s): 57afb09

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,10 +3,10 @@ import transformers
3
  from transformers import AutoModelWithLMHead, AutoTokenizer, pipeline
4
  from transformers import GPT2Tokenizer, GPT2Model
5
 
6
- model = GPT2Model.from_pretrained('LorenzoDeMattei/GePpeTto')
7
- tokenizer = GPT2Tokenizer.from_pretrained(
8
- 'LorenzoDeMattei/GePpeTto',
9
- )
10
 
11
  text_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('azaninello/GPT2-icc-new'),
12
  tokenizer='LorenzoDeMattei/GePpeTto',
 
3
  from transformers import AutoModelWithLMHead, AutoTokenizer, pipeline
4
  from transformers import GPT2Tokenizer, GPT2Model
5
 
6
+ #model = GPT2Model.from_pretrained('LorenzoDeMattei/GePpeTto')
7
+ #tokenizer = GPT2Tokenizer.from_pretrained(
8
+ #'LorenzoDeMattei/GePpeTto',
9
+ #)
10
 
11
  text_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('azaninello/GPT2-icc-new'),
12
  tokenizer='LorenzoDeMattei/GePpeTto',