Update app.py

app.py

@@ -31,10 +31,20 @@ pach_size = 256
 
 
 def predict_2(image):
-        
+        sizes_list = [256*10, 256*9, 256*8, 256*7, 256*6, 256*5, 256*4, 256*3, 256*2, 256*1]
+        min_size = min(image.shape)
+        if min_size%256==0:
+                size = min_size
+        else:
+            for i in sizes_list:
+                if min_size-i >= 0:
+                    size = i
+                    break
+            
+            
         image = Image.fromarray(image).resize((size,size))
         image = np.array(image)        
-        stride = 4
+        stride = 1
         steps = int(pach_size/stride)
         patches_img = patchify(image, (pach_size, pach_size, 3), step=steps)  #Step=256 for 256 patches means no overlap        
         patches_img = patches_img[:,:,0,:,:,:]