Update app.py

app.py

@@ -53,7 +53,13 @@ class DistillBERTClass(torch.nn.Module):
         pooler = self.dropout(pooler)
         output = self.classifier(pooler)
         return output
-model = torch.load("model.pt", map_location='cpu').eval()
+
+
+model = DistillBERTClass()
+optimizer = torch.optim.Adam(params =  model.parameters(), lr=LEARNING_RATE)
+checkpoint = torch.load("pytorch_distilbert_news.bin")
+model.load_state_dict(checkpoint['model'])
+optimizer.load_state_dict(checkpoint['opt']) 
 
 # print(model)
 # model = DistilBertForSequenceClassification.from_pretrained("model/distilbert-model1.pt", local_files_only=True)