Spaces:

neojex
/

LuxembourgishTextClassifier

Runtime error

neojex commited on Jun 1, 2023

Commit

af8cb67

•

1 Parent(s): 9704190

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ class SentimentClassifier(nn.Module):
         return self.out(output)
 # load the CNN binary classification model
 model = SentimentClassifier(len(class_names))
-model.load_state_dict(torch.load('./pytorch_model.bin'))
 tokenizer = BertTokenizer.from_pretrained('./')
 def encode(text):

         return self.out(output)
 # load the CNN binary classification model
 model = SentimentClassifier(len(class_names))
+model.load_state_dict(torch.load('./pytorch_model.bin', map_location=torch.device('cpu')))
 tokenizer = BertTokenizer.from_pretrained('./')
 def encode(text):