chore: adding spinners and caching

app.py

@@ -22,15 +22,18 @@ image, preprocessed_image = utils.load_image_from_url(
 )
 st.image(image, caption="Original Image")
 
-st.title("Attention Heat Maps")
-
 # Load the DINO model
-with st.spinner('Processing the heat maps. Wait for it 🔥...'):
-	dino = from_pretrained_keras("probing-vits/vit-dino-base16")
+@st.cache(persist=True)
+def load_dino():
+	return from_pretrained_keras("probing-vits/vit-dino-base16")
+
+dino = load_dino()
 
+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..."):
 	# 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])