Update app.py

app.py

@@ -6,17 +6,25 @@ from transformers import TrainingArguments, Trainer, AutoTokenizer, DataCollator
 from PIL import Image
 
 
-with open('labels.pickle', 'rb') as handle:
-    labels = pickle.load(handle)
+labels = {
+    "0":"biology"
+    "1":"computer science"
+    "2":"economics"
+    "3":"electrics"
+    "4":"finance"
+    "5":"math"
+    "6":"physics"
+    "7":"statistics"
+}
 
 def predict_topic_by_title_and_abstract(text, model):
     st.write(labels)
+    st.write(labels.values())
     tokens = tokenizer(text, return_tensors='pt', truncation=True)
     with torch.no_grad():
         logits = model(**tokens).logits
     probs = torch.nn.functional.softmax(logits[0], dim=0).numpy() * 100
-    #ans = list(zip(probs,labels.values()))
-    ans = list(zip(probs,[x for x in range(8)]))
+    ans = list(zip(probs,labels.values()))
     ans.sort(reverse=True)
     sum = 0
     i = 0