add authentication token to app.py

app.py

@@ -14,6 +14,9 @@ from utils import resize_image, make_image_square, cast_pil_image_to_torch_tenso
     remove_square_padding
 
 
+die_token = os.getenv("DIE_TOKEN")
+
+
 def die_inference(
     image_raw,
     num_of_die_iterations,
@@ -94,7 +97,13 @@ def main():
     ]
 
     # Load DIE model
-    die_model = UNetDIEModel(args=args)
+    die_token = os.getenv("DIE_TOKEN")
+    model_path = hf_hub_download(
+        repo_id="gabar92/DIE",
+        filename="2024_08_09_model_epoch_89.pt",
+        use_auth_token=die_token
+    )
+    die_model = UNetDIEModel(args=args, model_path=model_path)
 
     # Partially apply the model and device arguments to die_inference
     partial_die_inference = partial(die_inference, device=args.device, die_model=die_model)
@@ -122,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")