Update app.py

app.py

@@ -154,7 +154,7 @@ def create_parser():
     parser.add_argument(
         "--ddim_steps",
         type=int,
-        default=50,
+        default=2,
         help="number of ddim sampling steps",
     )
     parser.add_argument(
@@ -240,13 +240,13 @@ def create_parser():
     parser.add_argument(
         "--sd_ckpt",
         type=str,
-        default="/GPFS/data/ziyili/diffusion/a_final/model.ckpt",
+        default="stable_diffusion.ckpt",
         help="path to checkpoint of stable diffusion model",
     )
     parser.add_argument(
         "--grounding_ckpt",
         type=str,
-        default="/GPFS/data/ziyili/diffusion/a_final/checkpoint/checkpoint_latest.pth",
+        default="grounding_module.pth",
         help="path to checkpoint of grounding module",
     )
     parser.add_argument(
@@ -263,8 +263,6 @@ def create_parser():
         default="autocast"
     )
     opt = parser.parse_args()
-    
-    return opt
 
 
 def main():