Update pipeline.py

pipeline.py

@@ -1,7 +1,7 @@
 import torch
 import nltk
 import io
-import numpy as np
+import base64
 
 from pytorch_pretrained_biggan import BigGAN, one_hot_from_names, truncated_noise_sample
 
@@ -15,7 +15,7 @@ class PreTrainedPipeline():
         self.truncation = 0.1
 
 
-    def __call__(self, inputs: str) -> np.array:
+    def __call__(self, inputs: str) -> str:
         """
         Args:
             inputs (:obj:`str`):
@@ -35,4 +35,8 @@ class PreTrainedPipeline():
         with torch.no_grad():
             output = self.model(noise_vector, class_vector, self.truncation)
 
-        return output[0].numpy()
+        img = transforms.ToPILImage()(output[0])
+        buf = io.BytesIO()
+        img.save(buf, format="JPEG")
+
+        return base64.encodebytes(buf.getvalue()).decode('utf-8')