micrem73 commited on
Commit
ba58d54
·
1 Parent(s): 6e772e7

Add application file

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