change model access

app.py

@@ -111,11 +111,12 @@ def main():
         example_image_list.append([Image.open(image_path)])
 
     # Load DIE model
-    model_path = hf_hub_download(
-        repo_id="gabar92/DIE",
-        filename=args.die_model_path,
-        use_auth_token=die_token
-    )
+    # model_path = hf_hub_download(
+    #    repo_id="gabar92/DIE",
+    #    filename=args.die_model_path,
+    #    use_auth_token=die_token
+    #)
+    model_path = "./DIE/2024_08_09_model_epoch_89.pt"
     die_model = UNetDIEModel(args=args, model_path=model_path)
 
     # Partially apply the model and device arguments to die_inference