P01yH3dr0n commited on
Commit
1fd92f0
1 Parent(s): f0a3fb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -104,11 +104,11 @@ def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps
104
  i2i_image, i2i_str, i2i_noise
105
  )
106
  if not isinstance(img_data, bytes):
107
- return None, payload
108
  today_count += 1
109
  img = image_from_bytes(img_data)
110
-
111
- return img, payload
112
 
113
 
114
  def preview_ui():
 
104
  i2i_image, i2i_str, i2i_noise
105
  )
106
  if not isinstance(img_data, bytes):
107
+ return [None], payload
108
  today_count += 1
109
  img = image_from_bytes(img_data)
110
+ filename = str(today_count).rjust(5, '0') + '-' + str(payload['parameters']['seed']) + '.png'
111
+ return [(img, filename)], payload
112
 
113
 
114
  def preview_ui():