Upload processor

image_processing_efficientnet.py

@@ -1,4 +1,4 @@
-from transformers.image_processing_utils import BaseImageProcessor
+from transformers.image_processing_utils import BaseImageProcessor, BatchFeature
 from configuration_efficientnet import MODEL_NAMES
 from timm import create_model
 from timm.data import resolve_data_config
@@ -19,7 +19,8 @@ class EfficientNetImageProcessor(BaseImageProcessor):
 
     def preprocess(self, image):
         transforms = create_transform(**self.config)
-        return transforms(image).unsqueeze(0)
+        data = {'pixel_values': transforms(image).unsqueeze(0)}
+        return BatchFeature(data=data)
 
 __all__ = [
     "EfficientNetImageProcessor"