chore: removing cache

app.py

@@ -23,17 +23,14 @@ image, preprocessed_image = utils.load_image_from_url(
 st.image(image, caption="Original Image")
 
 # Load the DINO model
-@st.cache(persist=True)
-def load_dino():
-	return from_pretrained_keras("probing-vits/vit-dino-base16")
-
-dino = load_dino()
+with st.spinner("Loading the model..."):
+	dino = from_pretrained_keras("probing-vits/vit-dino-base16")
 
 with st.spinner("Generating the attention scores..."):
 	# Get the attention scores
 	_, attention_score_dict = dino.predict(preprocessed_image)
 
-with st.spinner("Generating the heat maps..."):
+with st.spinner("Generating the heat maps... HOLD ON!"):
 	# De-normalize the image for visual clarity. 
 	in1k_mean = tf.constant([0.485 * 255, 0.456 * 255, 0.406 * 255])  
 	in1k_std = tf.constant([0.229 * 255, 0.224 * 255, 0.225 * 255])