Update app.py

app.py

@@ -22,7 +22,7 @@ def get_models_and_tokenizers():
     model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=2)
     model.eval()
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model.load_state_dict(torch.load('./model.pth'))
+    model.load_state_dict(torch.load('./model.pth', map_location='cpu'))
 
     model_name_translator = "facebook/wmt19-ru-en"
     tokenizer_translator = FSMTTokenizer.from_pretrained(model_name_translator)