Upload processor

processor.py

@@ -20,12 +20,12 @@ class GIAProcessor(GitProcessor):
         }
         return result
 
-    def __call__(self, text=None, images=None, return_tensors=None, **kwargs):
-        if text is not None and images is None:
-            encoded_text = self.tokenizer(text, return_tensors=return_tensors)
+    def __call__(self, examples, return_tensors=None, **kwargs):
+        if "text" in examples and not "images" in examples:
+            encoded_text = self.tokenizer(examples["text"], return_tensors=return_tensors)
             encoding = self._group_texts(encoded_text)
-        elif text is not None and images is not None:
-            encoding = super().__call__(text, images, return_tensors, **kwargs)
+        elif "text" in examples and "images" in examples:
+            encoding = super().__call__(examples["text"], examples["images"], return_tensors, **kwargs)
 
         return encoding