Update app.py

app.py

@@ -122,7 +122,7 @@ def get_data(text):
 
 def send_to_img2img_def(image):
     result = Image.open(image)
-    return [result]
+    return result
 
 def send_to_txt2img(image):
 
@@ -175,7 +175,7 @@ def txt2img(prompt, negative_prompt, model, steps, sampler, cfg_scale, width, he
 
     job = prodia_client.wait(result)
 
-    return [job["imageUrl"]]
+    return job["imageUrl"]
 
 
 def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampler, cfg_scale, width, height, seed):
@@ -195,7 +195,7 @@ def img2img(input_image, denoising, prompt, negative_prompt, model, steps, sampl
 
     job = prodia_client.wait(result)
 
-    return [job["imageUrl"]]
+    return job["imageUrl"]
 
 
 css = """
@@ -244,7 +244,7 @@ with gr.Blocks(css=css) as demo:
                         seed = gr.Number(label="Seed", value=-1)
 
                 with gr.Column(scale=2):
-                    image_output = gr.Gallery(show_label=False, columns=4)
+                    image_output = gr.Image(show_label=False)
                     send_to_img2img = gr.Button(value="Send to img2img")
 
             
@@ -286,7 +286,7 @@ with gr.Blocks(css=css) as demo:
                         i2i_seed = gr.Number(label="Seed", value=-1)
 
                 with gr.Column(scale=2):
-                    i2i_image_output = gr.Gallery(show_label=False, columns=4)
+                    i2i_image_output = gr.Image(show_label=False)
     
             i2i_text_button.click(img2img, inputs=[i2i_image_input, i2i_denoising, i2i_prompt, i2i_negative_prompt,
                                                    model, i2i_steps, i2i_sampler, i2i_cfg_scale, i2i_width, i2i_height,