Update app.py

app.py

@@ -10,7 +10,7 @@ with open('labels.pickle', 'rb') as handle:
     labels = pickle.load(handle)
 
 # @st.cache
-def predict_topic_by_title_and_abstract(text):
+def predict_topic_by_title_and_abstract(text, model):
     tokens = tokenizer(text, return_tensors='pt', truncation=True)
     with torch.no_grad():
         logits = model(**tokens).logits
@@ -53,4 +53,4 @@ input_text = title + " " + abstract
 input_text = ''.join(filter(str.isalnum, input_text))
 
 if len(input_text.split()) > 0:
-    predict_topic_by_title_and_abstract(input_text)
+    predict_topic_by_title_and_abstract(input_text, model)