osanseviero
/

BigGAN-deep-128

Model card Files Files and versions Community

osanseviero HF staff commited on Aug 10, 2021

Commit

b5c7efb

•

1 Parent(s): abc03b4

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -0

pipeline.py CHANGED Viewed

@@ -36,5 +36,6 @@ class PreTrainedPipeline():
         img = transforms.ToPILImage()(output[0])
         buffer = io.BytesIO()
         img.save(buffer, format="JPEG")
         return buffer.getvalue()

         img = transforms.ToPILImage()(output[0])
         buffer = io.BytesIO()
         img.save(buffer, format="JPEG")
+        buffer.seek(0)
         return buffer.getvalue()