Update app.py

app.py

@@ -26,11 +26,12 @@ def bce_dice(y_true, y_pred):
     return bce(y_true, y_pred) - K.log(jacard(y_true, y_pred))
 
 
-size = 1024
+
 pach_size = 256
 
 
 def predict_2(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: