Update app.py

app.py

@@ -55,7 +55,19 @@ def double_res(input_image):
                 to_predict.append(lower_left_slice)
                 
 # predicting all images at once    
-    predicted = model.predict(np.array(to_predict), batch_size = 4)
+    completed = False
+    n = 16
+    while not completed:
+        try:
+            print("attempting with "+ str(n))
+            predicted = model.predict(np.array(to_predict),batch_size = n)
+            completed = True
+            print("completed with "+ str(n))
+        except:
+            print("attempt with " + str(n) + " failed")
+            n += -1
+            if n <= 0:
+                n = 1
     counter = 0
     for i in range(height):
         for j in range(width):