Spaces:

IvaElen
/

nlp_proj

Sleeping

IvaElen commited on Apr 27, 2023

Commit

f6324ba

1 Parent(s): 706fe96

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ def load_model():
         output_attentions = False,
         output_hidden_states = False
     )
-    model_finetuned.load_state_dict(torch.load('ErnieModel_imdb.pt', map_location = 'cpu'))
     tokenizer = transformers.AutoTokenizer.from_pretrained("nghuyong/ernie-2.0-base-en")
     return model_finetuned, tokenizer

         output_attentions = False,
         output_hidden_states = False
     )
+    model_finetuned.load_state_dict(torch.load('ErnieModel_imdb.pt', map_location=torch.device('cpu')))
     tokenizer = transformers.AutoTokenizer.from_pretrained("nghuyong/ernie-2.0-base-en")
     return model_finetuned, tokenizer