azaninello commited on
Commit
f466000
1 Parent(s): 91d7879

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ tokenizer = GPT2Tokenizer.from_pretrained(
8
  'LorenzoDeMattei/GePpeTto',
9
  )
10
 
11
- shroom_generator = pipeline("question-answering", model=AutoModelWithLMHead.from_pretrained('LorenzoDeMattei/GePpeTto'),
12
  tokenizer='LorenzoDeMattei/GePpeTto',
13
  do_sample=True,
14
  max_length=100,
 
8
  'LorenzoDeMattei/GePpeTto',
9
  )
10
 
11
+ shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('LorenzoDeMattei/GePpeTto'),
12
  tokenizer='LorenzoDeMattei/GePpeTto',
13
  do_sample=True,
14
  max_length=100,