Update app.py

app.py

@@ -167,9 +167,10 @@ def load_im(img):
             im1 = im.crop((left, top, right, bottom))
             im_box.append(im1)
             w+=new_w
-            yield im_box
+            yield im_box,[]
         h+=new_h
         w=0
+    yield im_box,im_box
 with gr.Blocks(css=css) as app:
     with gr.Row():
         with gr.Column():
@@ -186,8 +187,8 @@ with gr.Blocks(css=css) as app:
     out_html=gr.HTML()
     outp=gr.Gallery(columns=10)
     fingal=gr.Gallery(columns=10)
-    
-    im_btn.click(load_im,inp_im,outp)
-    go_btn = btn.click(run_dif,[inp,outp,model_drop,cnt],[fingal,out_html])
+    im_list=gr.Textbox()    
+    im_btn.click(load_im,inp_im,[outp,im_list])
+    go_btn = btn.click(run_dif,[inp,im_list,model_drop,cnt],[fingal,out_html])
     stop_btn.click(None,None,None,cancels=[go_btn])
 app.queue().launch()