Update test.py

test.py

@@ -33,6 +33,7 @@ def predict_hoax(title, content):
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 
+@st.cache_resource
 # LIME prediction function
 def predict_proba_for_lime(texts):
     results = []