Update app.py

app.py

@@ -28,10 +28,12 @@ def load_model():
     return tokenizer, model
 
 
-tokenizer, model = load_model()
 
 
-def run_model(model, tokenizer, title, summary):
+
+def run_model(title, summary):
+    tokenizer, model = load_model()
+    
     text = combine_title_summary(title, summary)
 
     tokens_info = tokenizer(
@@ -39,6 +41,7 @@ def run_model(model, tokenizer, title, summary):
         padding=False,
         truncation=True,
         return_tensors="pt",
+        max_length=512,
     )
 
     model.eval()
@@ -69,7 +72,7 @@ def main():
         if title == "" and abstract == "":
             st.error("At least one of title or abstract must be provided")
         else:
-            best_tags, best_probs = run_model(model, tokenizer, title, abstract)
+            best_tags, best_probs = run_model(title, abstract)
 
             df = pd.DataFrame(
                 dict(zip(best_tags, best_probs)).items(),