Update app.py

app.py

@@ -146,22 +146,23 @@ def load_im(img):
     im = Image.open(img)
     width, height = im.size
     new_w=int(width/10)
-    new_h=int(height/10)
-
-
-    
-    # Setting the points for cropped image
-    left = 0
-    top = 0
-    right = new_w
-    bottom = new_h
-     
-    # Cropped image of above dimension
-    # (It will not change original image)
-    im1 = im.crop((left, top, right, bottom))
-    im_box.append(im1)
-    return im_box
-
+    new_h=new_w
+    for i in range(height):
+        for b in range(width):
+              
+            # Setting the points for cropped image
+            left = b
+            top = i
+            right = left+new_w
+            bottom = top+new_h
+             
+            # Cropped image of above dimension
+            # (It will not change original image)
+            im1 = im.crop((left, top, right, bottom))
+            im_box.append(im1)
+            b+=new_w
+            yield im_box
+        i+=new_h
 
 with gr.Blocks(css=css) as app:
     with gr.Row():
@@ -175,7 +176,7 @@ with gr.Blocks(css=css) as app:
         model_drop=gr.Dropdown(label="Models", choices=models, type='index', value=models[0])
         cnt = gr.Number(value=1)
     out_html=gr.HTML()
-    outp=gr.Gallery()
+    outp=gr.Gallery(columns=10)
 
     im_btn.click(load_im,inp_im,outp)
     btn.click(run_dif,[inp,inp_im,model_drop,cnt],[outp,out_html])