Spaces:

probing-vits
/

attention-heat-maps

Running on T4

ariG23498 HF staff commited on Aug 5, 2022

Commit

0325baf

•

1 Parent(s): 321f9f3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -30,8 +30,8 @@ image, preprocessed_image = utils.load_image_from_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")

 	model_type="dino"
 )
 if uploaded_file:
+	image = Image.open(uploaded_file)
+	preprocessed_image = utils.preprocess_image(image, "dino")
 st.image(image, caption="Original Image")