Update app.py

app.py

@@ -23,7 +23,7 @@ bert_model = BertForSequenceClassification.from_pretrained('my_bert_model')
 tokenizer = BertTokenizer.from_pretrained('bert-base-multilingual-cased')
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 bert_model = bert_model.to(device)
-model_finetuned = torch.load('model.pt')
+model_finetuned = torch.load('model.pt', map_location=torch.device('cpu'))
 model_finetuned.eval()
 
 labels = ["не токсичный", "оскорбляющий", "непристойный", "угрожающий", "опасный"]