Update app.py

app.py

@@ -9,7 +9,6 @@ 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, model):
     tokens = tokenizer(text, return_tensors='pt', truncation=True)
     with torch.no_grad():
@@ -27,12 +26,6 @@ def predict_topic_by_title_and_abstract(text, model):
 
 tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
 
-#model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased", num_labels=8)
-#model.load_state_dict(
-#    torch.load(
-#        "./trained_model"
-#    )
-#)
 with open('trained_model.pickle', 'rb') as handle:
     model = pickle.load(handle)