update model path to use argparser

app.py

@@ -100,7 +100,7 @@ def main():
     die_token = os.getenv("DIE_TOKEN")
     model_path = hf_hub_download(
         repo_id="gabar92/DIE",
-        filename="2024_08_09_model_epoch_89.pt",
+        filename="args.die_model_path,
         use_auth_token=die_token
     )
     die_model = UNetDIEModel(args=args, model_path=model_path)
@@ -131,7 +131,7 @@ def parse_arguments():
 
     parser = argparse.ArgumentParser()
 
-    parser.add_argument("--die_model_path", default="/2024_08_09_model_epoch_89.pt")
+    parser.add_argument("--die_model_path", default="2024_08_09_model_epoch_89.pt")
     parser.add_argument("--device", default="cpu")
 
     parser.add_argument("--example_image_path", default="./example_images")