try4

app.py

@@ -14,7 +14,7 @@ Answer: [/INST]
 """
 
 # Load the model and tokenizer
-@st.cache_data(allow_output_mutation=True)
+@st.cache_data(persist="disk")
 def load_model():
     model_name = "walledai/walledguard-c"
     tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -22,7 +22,7 @@ def load_model():
     return tokenizer, model
 
 # Function to load image from URL
-@st.cache_resource(allow_output_mutation=True)
+@st.cache_resource(persist="disk")
 def load_image_from_url(url):
     response = requests.get(url)
     img = Image.open(BytesIO(response.content))