Update app.py

app.py

@@ -6,6 +6,9 @@ from transformers import TrainingArguments, Trainer, AutoTokenizer, DataCollator
 from PIL import Image
 
 
+with open('labels.pickle', 'rb') as handle:
+    labels = pickle.load(handle)
+
 # @st.cache
 def predict_topic_by_title_and_abstract(text):
     tokenized_text = tokenizer(text, return_tensors='pt')