Spaces:
Sleeping
Sleeping
aryrk
commited on
Commit
·
c9b5e9c
1
Parent(s):
4c9b339
[debug]
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|