cache the model so it doesn't reload for every image

app.py

@@ -4,6 +4,14 @@ from PIL import Image
 import streamlit as st
 import tensorflow as tf
 
+st.cache(show_spinner=True)
+def load_model():
+	# Load the DINO model
+	dino = from_pretrained_keras("probing-vits/vit-dino-base16")
+	return dino
+	
+dino=load_model()
+
 # Inputs
 st.title("Input your image")
 image_url = st.text_input(
@@ -27,10 +35,6 @@ if uploaded_file:
 	
 st.image(image, caption="Original Image")
 
-# Load the DINO model
-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)