[feat] resize output image to match input dimension

app.py

@@ -96,6 +96,11 @@ def reflection_removal(input_image, preprocess_type="resize_and_crop"):
             if file.startswith(input_filename) and file.endswith("_fake.png"):
                 result_path = os.path.join(root, file)
                 output_image = Image.open(result_path)
+                
+                if preprocess_type not in ["crop", "none"]:
+                    input_image = Image.open(input_image)
+                    output_image = output_image.resize(input_image.size)
+                
                 os.remove(result_path)
             elif file.startswith(input_filename) and file.endswith("_real.png"):
                 real_path = os.path.join(root, file)