Spaces:

BigSalmon
/

GPT2_Most_Probable

Runtime error

BigSalmon commited on Dec 11, 2021

Commit

11a45dd

1 Parent(s): 471111a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,6 @@ def get_model():
     tokenizer = AutoTokenizer.from_pretrained("gpt2")
     #model = AutoModelWithLMHead.from_pretrained("BigSalmon/MrLincoln12")
     model = AutoModelWithLMHead.from_pretrained("EleutherAI/gpt-neo-125M")
-    EleutherAI/gpt-neo-125M
     model.to(device)
     return model, tokenizer

     tokenizer = AutoTokenizer.from_pretrained("gpt2")
     #model = AutoModelWithLMHead.from_pretrained("BigSalmon/MrLincoln12")
     model = AutoModelWithLMHead.from_pretrained("EleutherAI/gpt-neo-125M")
     model.to(device)
     return model, tokenizer