[fix] update image result reader

app.py

@@ -26,11 +26,8 @@ if not os.path.exists(expected_model_path):
 def reflection_removal(input_images):
     for input_image in input_images:
         file_path = os.path.join(UPLOAD_DIR, os.path.basename(input_image.name))
-        print(f"Menyimpan file {input_image.name} ke {file_path}")
         shutil.copy(input_image.name, file_path)
             
-    print("Isi direktori uploaded_images:", os.listdir(UPLOAD_DIR))
-    print("Input images:", input_images)
     
     cmd = [
         "python", "test.py", 
@@ -47,7 +44,7 @@ def reflection_removal(input_images):
     output_images = []
     for root, _, files in os.walk(RESULTS_DIR):
         for file in files:
-            if file.endswith("_fake_B.png"):
+            if file.endswith("_fake.png"):
                 output_images.append(Image.open(os.path.join(root, file)))
 
     return output_images