Fix a bug in loading the input in tab_batch.

app.py

@@ -119,13 +119,13 @@ def predict(images, resolution, weights_file):
         # Apply the prediction mask to the original image
         image_pil = image_pil.resize(pred.shape[::-1])
         pred = np.repeat(np.expand_dims(pred, axis=-1), 3, axis=-1)
-        image_pred = (pred * np.array(image_pil)).astype(np.uint8)
+        image_masked = (pred * np.array(image_pil)).astype(np.uint8)
 
         torch.cuda.empty_cache()
 
         if tab_is_batch:
             save_file_path = os.path.join(save_dir, "{}.png".format(os.path.splitext(os.path.basename(image_src))[0]))
-            cv2.imwrite(save_file_path)
+            cv2.imwrite(save_file_path, image_masked)
             save_paths.append(save_file_path)
 
     if tab_is_batch:
@@ -134,7 +134,7 @@ def predict(images, resolution, weights_file):
             for file in save_paths:
                 zipf.write(file, os.path.basename(file))
 
-    return image, image_pred
+    return image, image_masked
 
 
 examples = [[_] for _ in glob('examples/*')][:]