Spaces:

azaninello
/

gpt2-general

Runtime error

azaninello commited on Jun 27, 2022

Commit

ea0bcc0

•

1 Parent(s): 2b30cc1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ tokenizer = GPT2Tokenizer.from_pretrained(
     'LorenzoDeMattei/GePpeTto',
 )
-shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('LorenzoDeMattei/GePpeTto'),
     tokenizer='LorenzoDeMattei/GePpeTto',
     do_sample=True,
     max_length=100,

     'LorenzoDeMattei/GePpeTto',
 )
+shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('azaninello/GPT2-icc'),
     tokenizer='LorenzoDeMattei/GePpeTto',
     do_sample=True,
     max_length=100,