Update pipeline.py

pipeline.py

@@ -25,14 +25,14 @@ class PreTrainedPipeline():
             output_ids = self.model.generate(pixel_values, **self.gen_kwargs).sequences
             return output_ids
 
+        self.generate = _generate
+
         # compile the model
         image_path = os.path.join(path, 'val_000000039769.jpg')
         image = Image.open(image_path)
         self(image)
         image.close()
-
-        self.generate = _generate
-
+        
     def __call__(self, inputs: "Image.Image") -> List[str]:
         """
         Args: