Spaces:

aryrk
/

SingleImageReflectionRemoval

Sleeping

aryrk commited on Dec 2, 2024

Commit

af1a755

1 Parent(s): 0eb4294

[debug]

Files changed (1) hide show

app.py CHANGED Viewed

@@ -44,7 +44,7 @@ def reflection_removal(input_image):
         "--model", "test", "--netG", "unet_256",
         "--direction", "AtoB", "--dataset_mode", "single",
         "--norm", "batch", "--epoch", "310",
-        "--num_test", count_files(UPLOAD_DIR),
         "--gpu_ids", "-1"
     ]
     subprocess.run(cmd, check=True)

         "--model", "test", "--netG", "unet_256",
         "--direction", "AtoB", "--dataset_mode", "single",
         "--norm", "batch", "--epoch", "310",
+        "--num_test", str(count_files(UPLOAD_DIR)),
         "--gpu_ids", "-1"
     ]
     subprocess.run(cmd, check=True)