Update app.py

app.py

@@ -70,8 +70,18 @@ def double_res(input_image):
                                                         128-64:i*128+64, j*128-64:j*128+64]/255
                 to_predict.append(lower_left_slice)
                 
-# predicting all images at once    
-    predicted = model.predict(np.array(to_predict),batch_size = 4)
+# predicting all images at once
+    completed = False
+    n = 8
+    while not completed:
+        try:
+            predicted = model.predict(np.array(to_predict),batch_size = n)
+            completed = True
+            print("completed with "+ str(n))
+        except:
+            n += -1
+            if n <= 0:
+                n = 1
     counter = 0
     for i in range(height):
         for j in range(width):