Update app.py

app.py

@@ -59,10 +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", map_location=torch.device('cpu'))
-model.load_state_dict(checkpoint['model'])
-optimizer.load_state_dict(checkpoint['opt']) 
-model.to("cpu")
+model = torch.load("pytorch_distilbert_news.bin", map_location=torch.device('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)