P01yH3dr0n commited on
Commit
71b400a
1 Parent(s): 9af776c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -84,11 +84,11 @@ def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps
84
  smea, dyn, dyn_threshold, cfg_rescale
85
  )
86
  if not isinstance(img_data, bytes):
87
- return gr.Image(value=None, height=100), payload
88
  today_count += 1
89
  img = image_from_bytes(img_data)
90
 
91
- return gr.Image(value=img, height=height*721//width), payload
92
 
93
 
94
  def preview_ui():
 
84
  smea, dyn, dyn_threshold, cfg_rescale
85
  )
86
  if not isinstance(img_data, bytes):
87
+ return None, payload
88
  today_count += 1
89
  img = image_from_bytes(img_data)
90
 
91
+ return img, payload
92
 
93
 
94
  def preview_ui():