aryrk commited on
Commit
c9b5e9c
·
1 Parent(s): 4c9b339
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -32,6 +32,7 @@ def reflection_removal(input_image):
32
  shutil.copy(input_image, file_path)
33
 
34
  input_filename = os.path.splitext(os.path.basename(file_path))[0]
 
35
 
36
  cmd = [
37
  "python", "test.py",
@@ -47,7 +48,11 @@ def reflection_removal(input_image):
47
 
48
  for root, _, files in os.walk(RESULTS_DIR):
49
  for file in files:
50
- if file.endswith("_fake.png") and file.startswith(input_filename):
 
 
 
 
51
  result_path = os.path.join(root, file)
52
  return [Image.open(result_path)]
53
 
 
32
  shutil.copy(input_image, file_path)
33
 
34
  input_filename = os.path.splitext(os.path.basename(file_path))[0]
35
+ print(f"Processing {input_filename}...")
36
 
37
  cmd = [
38
  "python", "test.py",
 
48
 
49
  for root, _, files in os.walk(RESULTS_DIR):
50
  for file in files:
51
+ print(os.path.join(root, file))
52
+
53
+ for root, _, files in os.walk(RESULTS_DIR):
54
+ for file in files:
55
+ if file.startswith(input_filename) and file.endswith("_fake.png"):
56
  result_path = os.path.join(root, file)
57
  return [Image.open(result_path)]
58