Update app.py

app.py

@@ -3,18 +3,25 @@ import streamlit as st
 
 
 @st.cache_data
-def prepare():
+def prepare_model():
+    """
+    Prepare the tokenizer and the model for classification.
+    """
     tokenizer = AutoTokenizer.from_pretrained("oracat/bert-paper-classifier")
     model = AutoModelForSequenceClassification.from_pretrained("oracat/bert-paper-classifier")
+    return (tokenizer, model)
 
 
 def process(text):
+    """
+    Translate incoming text to tokens and classify it
+    """
     pipe = pipeline("text-classification", model=model, tokenizer=tokenizer)
     result = pipe(text)[0]
     return results['label']
 
 
-prepare()
+tokenizer, model = prepare_model()
 
 st.markdown("### Hello, paper classifier!")