Update app.py

app.py

@@ -87,7 +87,16 @@ def predict_2(image):
 from empatches import EMPatches
 
 def predict_patches(image):
-        #img = Image.open(image)
+        size = 1024
+        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)/255
         emp = EMPatches()