Update app.py

app.py

@@ -10,6 +10,7 @@ with open('labels.pickle', 'rb') as handle:
     labels = pickle.load(handle)
 
 def predict_topic_by_title_and_abstract(text, model):
+    st.write(labels)
     tokens = tokenizer(text, return_tensors='pt', truncation=True)
     with torch.no_grad():
         logits = model(**tokens).logits