Update app.py

app.py

@@ -27,7 +27,7 @@ class BertClass(torch.nn.Module):
 bert_path = "bert-base-uncased"
 bert_tokenizer = AutoTokenizer.from_pretrained(bert_path)
 bert_model = BertForSequenceClassification.from_pretrained(bert_path, num_labels=6)
-tuned_model = model = torch.load("pytorch_bert_toxic.bin")
+tuned_model = model = torch.load("pytorch_bert_toxic.bin", map_location=torch.device("cpu"))
 
 # Read and format data.
 tweets_raw = pd.read_csv("test.csv", nrows=20)