Add stage1 and stage2 configuration options to argument parser in app.py for enhanced flexibility in model configuration.

app.py

@@ -118,6 +118,18 @@ def gen_image(input_image, seed, scale, step):
 
 parser = argparse.ArgumentParser()
 
+parser.add_argument(
+    "--stage1_config",
+    type=str,
+    default="configs/nf7_v3_SNR_rd_size_stroke.yaml",
+    help="config for stage1",
+)
+parser.add_argument(
+    "--stage2_config",
+    type=str,
+    default="configs/stage2-v2-snr.yaml",
+    help="config for stage2",
+)
 parser.add_argument("--device", type=str, default="cuda")
 args = parser.parse_args()