Update app.py

app.py

@@ -59,9 +59,10 @@ model = DistillBERTClass()
 LEARNING_RATE = 1e-05
 
 optimizer = torch.optim.Adam(params =  model.parameters(), lr=LEARNING_RATE)
-checkpoint = torch.load("pytorch_distilbert_news.bin")
+checkpoint = torch.load("pytorch_distilbert_news.bin").to("cpu")
 model.load_state_dict(checkpoint['model'])
 optimizer.load_state_dict(checkpoint['opt']) 
+model.to("cpu")
 
 # print(model)
 # model = DistilBertForSequenceClassification.from_pretrained("model/distilbert-model1.pt", local_files_only=True)