osanseviero HF staff commited on
Commit
5735fa4
1 Parent(s): 699e3bc

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -21,7 +21,7 @@ class PreTrainedPipeline():
21
  inputs (:obj:`str`):
22
  a string containing some text
23
  Return:
24
- A :obj:`PIL.Image`. The raw image representation as PIL.
25
  """
26
  class_vector = one_hot_from_names([inputs], batch_size=1)
27
  if type(class_vector) == type(None):
 
21
  inputs (:obj:`str`):
22
  a string containing some text
23
  Return:
24
+ A :obj:`str`. A base64 string representing the image.
25
  """
26
  class_vector = one_hot_from_names([inputs], batch_size=1)
27
  if type(class_vector) == type(None):