update model path parameter in argument parser

app.py

@@ -96,14 +96,14 @@ def main():
         for image_path in os.listdir(args.example_image_path)
     ]
 
-
     # Load DIE model
-    model_path = hf_hub_download(
+    args.die_model_path = hf_hub_download(
         repo_id="gabar92/die",
         filename=args.die_model_path,
         use_auth_token=die_token
     )
-    die_model = UNetDIEModel(args=args, model_path=model_path)
+    
+    die_model = UNetDIEModel(args=args)
 
     # Partially apply the model and device arguments to die_inference
     partial_die_inference = partial(die_inference, device=args.device, die_model=die_model)