Update app.py

app.py

@@ -38,9 +38,16 @@ def infer(prompt, init_image):
     with torch.no_grad():
         torch.cuda.empty_cache()
     
+    outputs = []
+    
     res = pipe(alpha=1)
-
-    return res.images[0]
+    outputs.append(res.images[0])
+    res = pipe(alpha=1.5)
+    outputs.append(res.images[0])
+    res = pipe(alpha=2)
+    outputs.append(res.images[0])
+    
+    return outputs
     #return 'trained success'
 
 title = """
@@ -115,11 +122,11 @@ with gr.Blocks(css=css) as block:
         
         submit_btn = gr.Button("Train")
         
-        image_output = gr.Image(label="Edited image")
+       gallery_output = gr.Gallery(label="Edited images")
         #text_output = gr.Image(label="trained status")
         
         gr.HTML(article)
 
-    submit_btn.click(fn=infer, inputs=[prompt_input,image_init], outputs=[image_output])
+    submit_btn.click(fn=infer, inputs=[prompt_input,image_init], outputs=[gallery_output])
     
 block.queue(max_size=12).launch(show_api=False)