fvonhoven commited on
Commit
77b73e5
1 Parent(s): 0d4e67f

Alas, return 1 image

Browse files
Files changed (1) hide show
  1. handler.py +1 -2
handler.py CHANGED
@@ -77,7 +77,6 @@ class EndpointHandler():
77
  prompt = data.pop("inputs", None)
78
  image = data.pop("image", None)
79
  controlnet_type = data.pop("controlnet_type", None)
80
- count = data.pop("count", 1)
81
 
82
  # Check if neither prompt nor image is provided
83
  if prompt is None and image is None:
@@ -120,7 +119,7 @@ class EndpointHandler():
120
 
121
 
122
  # return first generate PIL image
123
- return out.images
124
 
125
  # helper to decode input image
126
  def decode_base64_image(self, image_string):
 
77
  prompt = data.pop("inputs", None)
78
  image = data.pop("image", None)
79
  controlnet_type = data.pop("controlnet_type", None)
 
80
 
81
  # Check if neither prompt nor image is provided
82
  if prompt is None and image is None:
 
119
 
120
 
121
  # return first generate PIL image
122
+ return out.images[0]
123
 
124
  # helper to decode input image
125
  def decode_base64_image(self, image_string):