Spaces:

walledai
/

walledai-walledguard-c

Running

RishabhBhardwaj commited on 8 days ago

Commit

2ec2072

•

1 Parent(s): 6507278

try5

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ def load_model():
     return tokenizer, model
 # Function to load image from URL
-@st.cache_resource(persist="disk")
 def load_image_from_url(url):
     response = requests.get(url)
     img = Image.open(BytesIO(response.content))

     return tokenizer, model
 # Function to load image from URL
+@st.cache_resource()
 def load_image_from_url(url):
     response = requests.get(url)
     img = Image.open(BytesIO(response.content))