Update app.py

app.py

@@ -118,10 +118,10 @@ def run_dif_color(out_prompt,im_path,model_drop,tint,im_height,im_width):
                     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')                
-                        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 = get_concat_h_cut(f'comb-{uid}-tmp.png',f'tmp-{uid}.png')
-                            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[cnt]}')
+                        #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