Spaces:

gefedya
/

YSDA_transfprmers_nlp_ops

Runtime error

gefedya commited on Apr 17, 2023

Commit

5869fc6

1 Parent(s): 5ac7ad7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import torch
 @st.cache()
 def get_model():
     model = AutoModelForSequenceClassification.from_pretrained("siebert/sentiment-roberta-large-english", num_labels=2)
-    model.load_state_dict(torch.load('cached_model.pth'))
     return model
 @st.cache()

 @st.cache()
 def get_model():
     model = AutoModelForSequenceClassification.from_pretrained("siebert/sentiment-roberta-large-english", num_labels=2)
+    model.load_state_dict(torch.load('cached_model.pth', map_location=torch.device('cpu')))
     return model
 @st.cache()