Spaces:

SatAT
/

transformer_service

Sleeping

SatAT commited on Apr 15, 2023

Commit

a348a05

1 Parent(s): a557725

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -58,7 +58,7 @@ model = BertForSequenceClassification.from_pretrained(
     output_attentions = False, # Whether the model returns attentions weights.
     output_hidden_states = False, # Whether the model returns all hidden-states.
 )
-model.load_state_dict(torch.load("model_last_version.pt"))
 # model.to(device)
 model.eval()
 with torch.no_grad():

     output_attentions = False, # Whether the model returns attentions weights.
     output_hidden_states = False, # Whether the model returns all hidden-states.
 )
+model.load_state_dict(torch.load("model_last_version.pt", map_location=torch.device('cpu')))
 # model.to(device)
 model.eval()
 with torch.no_grad():