Update pipeline.py

pipeline.py

@@ -1,4 +1,5 @@
 import os
+from typing import Dict, List, Any
 from PIL import Image
 import jax
 from transformers import ViTFeatureExtractor, AutoTokenizer, FlaxVisionEncoderDecoderModel
@@ -30,7 +31,7 @@ class PreTrainedPipeline():
         output_ids = self.model.generate(pixel_values, **self.gen_kwargs).sequences
         return output_ids
 
-    def __call__(self, inputs: "Image.Image") -> List[Dict[str, Any]]:
+    def __call__(self, inputs: "Image.Image") -> List[str]:
         """
         Args:
         Return:
@@ -42,4 +43,4 @@ class PreTrainedPipeline():
         preds = self.tokenizer.batch_decode(output_ids, skip_special_tokens=True)
         preds = [pred.strip() for pred in preds]
 
-        return preds[0]
+        return preds