[fix] add try attemp

app.py

@@ -39,9 +39,7 @@ def clear_session_files(session_id):
         shutil.rmtree(session_dir)
 
 def reflection_removal(input_image, preprocess_type="resize_and_crop"):
-    if preprocess_type == "":
-        preprocess_type = "resize_and_crop"
-    
+
     session_id = generate_session_id()
     session_dir = os.path.join(UPLOAD_DIR, session_id)
     upload_dir = os.path.join(session_dir, "uploads")
@@ -67,7 +65,25 @@ def reflection_removal(input_image, preprocess_type="resize_and_crop"):
         "--gpu_ids", "-1",
         "--preprocess", preprocess_type
     ]
-    subprocess.run(cmd, check=True)
+    attempt = 0
+    while True:
+        attempt += 1
+        try:
+            subprocess.run(cmd, check=True)
+            break
+        except subprocess.CalledProcessError as e:
+            cmd = [
+                "python", "test.py",
+                "--dataroot", upload_dir,
+                "--name", "SingleImageReflectionRemoval",
+                "--model", "test", "--netG", "unet_256",
+                "--direction", "AtoB", "--dataset_mode", "single",
+                "--norm", "batch", "--epoch", "310",
+                "--num_test", "1",
+                "--gpu_ids", "-1",
+            ]
+            if attempt > 2:
+                return "No results found. Please try again with a different image."
 
     output_image = None
     for root, _, files in os.walk(RESULTS_DIR):
@@ -114,7 +130,6 @@ iface = gr.Interface(
     ],
     title="Reflection Remover with Pix2Pix",
     description="Upload images to remove reflections using a Pix2Pix model. You can also try the sample images below.",
-    live=True
 )
 
 if __name__ == "__main__":