Update app.py

app.py

@@ -104,16 +104,16 @@ def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps
         i2i_image, i2i_str, i2i_noise
     )
     if not isinstance(img_data, bytes):
-        return [None], payload
+        return None, payload
     today_count += 1
     img = image_from_bytes(img_data)
     filename = str(today_count).rjust(5, '0') + '-' + str(payload['parameters']['seed']) + '.png'
-    return [(img, filename)], payload
+    return gr.Image(value=img, label=filename), payload
 
 
 def preview_ui():
     with gr.Blocks(css='#preview_image { height: 100%;}'):
-        image = gr.Gallery(elem_id='preview_image', interactive=False, type='pil')
+        image = gr.Image(elem_id='preview_image', interactive=False, type='pil')
         info = gr.JSON(value={}, label="Submitted Payload")
     return image, info