Update app.py

app.py

@@ -34,7 +34,8 @@ def get_top95(y_predict, convert_target):
     return lst_labels
 #
 # model = MyModel()
-model = torch.load("distilbert-model1.pt", map_location='cpu').eval()
+model = torch.load("pytorch_distilbert_news.bin", map_location='cpu').eval()
+
 # print(model)
 # model = DistilBertForSequenceClassification.from_pretrained("model/distilbert-model1.pt", local_files_only=True)
 # tokenizer = BigBirdTokenizer.from_pretrained('google/bigbird-pegasus-large-arxiv')