Update utils.py

utils.py

@@ -11,7 +11,7 @@ def get_most_probability_terms(logits, top_k=5):
   return indices[:, :top_k]
 
 @st.cache(suppress_st_warning=True)
-def predict(text, model, tokenizer, my_linear, top_k=3):
+def predict(text, model, tokenizer, my_linear, top_k=10):
   tokens = tokenizer.encode(text)
   with torch.no_grad():
     outputs = model(torch.as_tensor([tokens]))[0]
@@ -19,8 +19,9 @@ def predict(text, model, tokenizer, my_linear, top_k=3):
   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):
-  codes = predict(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]:
     answer += [code + ": " + classes_desc[code]]