Update app.py

app.py

@@ -27,12 +27,14 @@ def predict_topic_by_title_and_abstract(text):
 
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 
-model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
-model.load_state_dict(
-    torch.load(
-        "./trained_model"
-    )
-)
+#model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
+#model.load_state_dict(
+#    torch.load(
+#        "./trained_model"
+#    )
+#)
+with open('trained_model.pickle', 'rb') as handle:
+    model = pickle.load(handle)
 
 image = Image.open('logo.png')