Update pipeline.py

pipeline.py

@@ -34,8 +34,4 @@ class PreTrainedPipeline():
 
         # Ugly hack, I'm sure there's a better way of doing what I want.
         img = transforms.ToPILImage()(output[0])
-        buffer = io.BytesIO()
-        img.save(buffer, format="JPEG")
-        buffer.seek(0)
-
-        return buffer.getvalue()
+        return img