Update app.py

app.py

@@ -42,8 +42,8 @@ def get_attention_maps(pixel_values, attentions, nh):
 
   return attention_maps
 
-feature_extractor = ViTFeatureExtractor.from_pretrained("facebook/dino-vitb16", do_resize=False)
-model = ViTModel.from_pretrained("facebook/dino-vitb16", add_pooling_layer=False)
+feature_extractor = ViTFeatureExtractor.from_pretrained("facebook/dino-vitb8", do_resize=False)
+model = ViTModel.from_pretrained("facebook/dino-vitb8", add_pooling_layer=False)
 
 def visualize_attention(image):
   # normalize channels