Update utils.py

utils.py

@@ -10,7 +10,6 @@ def get_most_probability_terms(logits, top_k=5):
   _, indices = torch.sort(logits, dim=1, descending=True)
   return indices[:, :top_k]
 
-@st.cache(suppress_st_warning=True)
 def predict(text, model, tokenizer, my_linear, top_k=10):
   tokens = tokenizer.encode(text)
   with torch.no_grad():
@@ -19,8 +18,7 @@ def predict(text, model, tokenizer, my_linear, top_k=10):
   return np.array(classes)[get_most_probability_terms(logits, top_k).cpu().numpy()][0]
 
 
-@st.cache(suppress_st_warning=True)
-def get_answer_with_desc(text, model, tokenizer, my_linear, top_k=3):
+def get_answer_with_desc(text, model, tokenizer, my_linear, top_k=10):
   codes = predict(text, model, tokenizer, my_linear)
   answer = ["Possible text topics:"]
   for code in codes[:top_k]: