Update app.py

app.py

@@ -115,15 +115,15 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
                 print(f'out_box:: {out_box}')
                 if out_box:
                     if len(out_box)>1:
-                        im_roll = get_concat_v_cut(f'{out_box[0]}',f'{out_box[1]}')
-                        im_roll.save(f'comb-{uid}-tmp.png')                
+                        #im_roll = get_concat_v_cut(f'{out_box[0]}',f'{out_box[1]}')
+                        #im_roll.save(f'comb-{uid}-tmp.png')                
                         for i in range(2,len(out_box)):
                             im_roll = get_concat_h_cut(f'comb-{uid}-tmp.png',f'{out_box[i]}')
                             im_roll.save(f'comb-{uid}-tmp.png')
                         out = f'comb-{uid}-tmp.png'
                     else:
-                        #tmp_im = Image.open(out_box[0])
-                        tmp_im = out_box[0]
+                        tmp_im = Image.open(out_box[0])
+                        #tmp_im = out_box[0]
                         tmp_im.save(f'comb-{uid}-tmp.png')
                         out = f'comb-{uid}-tmp.png'                   
                     yield out,out_html