add file uploader (#1)

- add file uploader (19e4e13690edc6892f4f5c2cdd8ffb996a24ebc8)
- cache the model so it doesn't reload for every image (bc751867842b5ed8f9a682576791de8c80c2f834)


Co-authored-by: Ceyda Cinarel <[email protected]>

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(
@@ -11,6 +19,7 @@ image_url = st.text_input(
 	value="https://dl.fbaipublicfiles.com/dino/img.png",
 	placeholder="https://your-favourite-image.png"
 )
+uploaded_files = st.file_uploader("or an image file", type =["jpg","jpeg"])
 
 # Outputs
 st.title("Original Image from URL")
@@ -20,12 +29,12 @@ image, preprocessed_image = utils.load_image_from_url(
 	image_url,
 	model_type="dino"
 )
+if uploaded_file:
+	image = Image.open(im)
+	preprocessed_image = utils.preprocess_image(image, model_type)
+	
 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)