Update utils.py

utils.py

@@ -1,4 +1,5 @@
 import torch
+import streamlit as st
 
 
 def get_text(title: str, abstract: str):
@@ -29,5 +30,8 @@ def get_labels(text, model, tokenizer, count_labels=8):
       if cumsum > 0.95 and len(result_labels) >= 1:
           return result_labels
       result_labels.append(pair[1])
-  
+ 
+ @st.cache
+ def load_model(model, filename):
+   return model.load_state_dict(torch.load(filename))