switch classifier google/vit-base-patch16-224

app.py

@@ -23,9 +23,9 @@ feature_extractor = ViTFeatureExtractor()
 #feature_extractor = ViTFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
 
 #the following gets called by classify_image() 
-feature_extractor = PerceiverFeatureExtractor.from_pretrained("deepmind/vision-perceiver-conv")
-model = PerceiverForImageClassificationConvProcessing.from_pretrained("deepmind/vision-perceiver-conv")
-
+feature_extractor = PerceiverFeatureExtractor.from_pretrained("google/vit-base-patch16-224")
+model = PerceiverForImageClassificationConvProcessing.from_pretrained("google/vit-base-patch16-224")
+#google/vit-base-patch16-224, deepmind/vision-perceiver-con
 image_pipe = ImageClassificationPipeline(model=model, feature_extractor=feature_extractor)