Update app.py

app.py

@@ -8,14 +8,7 @@ torch.hub.download_url_to_file('https://storage.googleapis.com/perceiver_io/dalm
 feature_extractor = PerceiverFeatureExtractor()
 model = PerceiverForImageClassificationConvProcessing.from_pretrained("deepmind/vision-perceiver-conv")
 
-# define custom pipeline as Perceiver expects "inputs" rather than "pixel_values"
-class CustomPipeline(ImageClassificationPipeline):
-    def _forward(self, model_inputs):
-        inputs = model_inputs["pixel_values"]
-        model_outputs = self.model(inputs=inputs)
-        return model_outputs
-
-image_pipe = CustomPipeline(model=model, feature_extractor=feature_extractor)
+image_pipe = ImageClassificationPipeline(model=model, feature_extractor=feature_extractor)
 
 def classify_image(image):
   results = image_pipe(image)