Spaces:

azaninello
/

gpt2-general

Runtime error

azaninello commited on May 18, 2022

Commit

a8d925c

•

1 Parent(s): ca9faff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,8 @@
 import gradio as gr
 import transformers
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
-shroom_generator = pipeline("text-generation", model=AutoModelForCausalLM.from_pretrained('LorenzoDeMattei/GePpeTto'),
     tokenizer='LorenzoDeMattei/GePpeTto',
     max_length=50)

 import gradio as gr
 import transformers
+from transformers import AutoModelWithLMHead, AutoTokenizer, pipeline
+shroom_generator = pipeline("text-generation", model=AutoModelWithLMHead.from_pretrained('LorenzoDeMattei/GePpeTto'),
     tokenizer='LorenzoDeMattei/GePpeTto',
     max_length=50)