Update app.py

app.py

@@ -123,7 +123,7 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
                         #im_roll = get_concat_h_cut(f'comb-{uid}-tmp.png',f'tmp-{uid}.png')
                         im_roll.save(f'comb-{uid}-tmp.png')
                         out = f'comb-{uid}-tmp.png'
-                        yield out,out_html
+                        yield gr.Image(out),out_html
                         
                     else:
                         tmp_im = Image.open(out_box[0])