Spaces:

probing-vits
/

class-attention-map

Runtime error

sayakpaul HF staff commited on May 2, 2022

Commit

1c951f1

•

1 Parent(s): 98e527d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ _MODEL = get_model()
 def show_plot(image):
     """Function to be called when user hits submit on the UI."""
     _, preprocessed_image = utils.preprocess_image(
-        image, "deit_tiny_patch16_224"
     )
     _, _, ca_atn_score_dict = _MODEL.predict(preprocessed_image)

 def show_plot(image):
     """Function to be called when user hits submit on the UI."""
     _, preprocessed_image = utils.preprocess_image(
+        image, _RESOLUTION
     )
     _, _, ca_atn_score_dict = _MODEL.predict(preprocessed_image)