[feat] update load size and crop size

app.py

@@ -42,13 +42,16 @@ def reflection_removal(input_image):
     
     input_filename = os.path.splitext(os.path.basename(file_path))[0]
     
-    aspect_ratio = 1.0
+    load_size = 286
+    crop_size = 256
     
     try:
         with Image.open(file_path) as img:
             width, height = img.size
-            aspect_ratio = width / height
-            print(f"Aspect Ratio: {aspect_ratio}")
+            load_size = max(width, height)
+            crop_size = min(width, height)
+            print(f"Image Dimensions: {width}x{height}")
+            print(f"load_size: {load_size}, crop_size: {crop_size}")
     except Exception as e:
         pass
     
@@ -61,7 +64,8 @@ def reflection_removal(input_image):
         "--norm", "batch", "--epoch", "310",
         "--num_test", str(count_files(UPLOAD_DIR)),
         "--gpu_ids", "-1",
-        "--aspect_ratio", str(aspect_ratio)
+        "--load_size", str(load_size),
+        "--crop_size", str(crop_size),
     ]
     subprocess.run(cmd, check=True)