Add application file
Browse files
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,
|