osanseviero
/

BigGAN-deep-128

Model card Files Files and versions Community

osanseviero commited on Aug 10, 2021

Commit

55fd70d

•

1 Parent(s): 8b86a1f

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -21,7 +21,7 @@ class PreTrainedPipeline():
             inputs (:obj:`str`):
                 a string containing some text
         Return:
-            A :obj:`str`. A base64 string representing the image.
         """
         class_vector = one_hot_from_names([inputs], batch_size=1)
         if type(class_vector) == type(None):

             inputs (:obj:`str`):
                 a string containing some text
         Return:
+            A :obj:`PIL.Image` with the raw image representation as PIL.
         """
         class_vector = one_hot_from_names([inputs], batch_size=1)
         if type(class_vector) == type(None):