micrem73 commited on
Commit
5dc70b2
1 Parent(s): f0dd1e6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,11 +5,11 @@ from transformers import GPT2Tokenizer, GPT2Model
5
 
6
  model = GPT2Model.from_pretrained('micrem73/GePpeTto-finetuned-ricettetrentine')
7
  tokenizer = GPT2Tokenizer.from_pretrained(
8
- 'micrem73/GePpeTto-finetuned-ricettetrentine',
9
  )
10
 
11
  shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('micrem73/GePpeTto-finetuned-ricettetrentine'),
12
- tokenizer='micrem73/GePpeTto-finetuned-ricettetrentine',
13
  do_sample=True,
14
  max_length=100,
15
  top_k=50,
 
5
 
6
  model = GPT2Model.from_pretrained('micrem73/GePpeTto-finetuned-ricettetrentine')
7
  tokenizer = GPT2Tokenizer.from_pretrained(
8
+ 'LorenzoDeMattei/GePpeTto',
9
  )
10
 
11
  shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('micrem73/GePpeTto-finetuned-ricettetrentine'),
12
+ tokenizer='LorenzoDeMattei/GePpeTto',
13
  do_sample=True,
14
  max_length=100,
15
  top_k=50,