Update app.py

app.py

@@ -59,7 +59,7 @@ model = DistillBERTClass()
 LEARNING_RATE = 1e-05
 
 optimizer = torch.optim.Adam(params =  model.parameters(), lr=LEARNING_RATE)
-model = torch.load("pytorch_distilbert_news (1).bin", map_location=torch.device('cpu'))
+model = torch.load("pytorch_distilbert_news (3).bin", map_location=torch.device('cpu'))
 # model.load_state_dict(checkpoint['model'])
 # optimizer.load_state_dict(checkpoint['opt']) 
 # model.to("cpu")
@@ -91,7 +91,7 @@ def get_predict(text):
     )
     logits = outputs[0]
     y_predict = torch.nn.functional.softmax(logits).cpu().detach().numpy()
-    file_path = "decode_target (1).json"
+    file_path = "sample.json"
 
     with open(file_path, 'r') as json_file:
         decode_target = json.load(json_file)