Spaces:

BigSalmon
/

GPT2_Most_Probable

Runtime error

BigSalmon commited on Feb 11, 2022

Commit

4d5c512

1 Parent(s): 623501a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,6 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 def get_model():
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln21")
     model = AutoModelWithLMHead.from_pretrained("BigSalmon/InformalToFormalLincoln21")
-    model.to(device)
     return model, tokenizer
 model, tokenizer = get_model()

 def get_model():
     tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln21")
     model = AutoModelWithLMHead.from_pretrained("BigSalmon/InformalToFormalLincoln21")
     return model, tokenizer
 model, tokenizer = get_model()