bugfix

app.py

@@ -15,9 +15,6 @@ 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,
@@ -91,6 +88,8 @@ def main():
 
     num_of_die_iterations_list = [1, 2, 3]
 
+    die_token = os.getenv("DIE_TOKEN")
+
     # Load example images from the private repository
     example_image_list = []
     example_images = [
@@ -112,7 +111,6 @@ def main():
         example_image_list.append([Image.open(image_path)])
 
     # Load DIE model
-    die_token = os.getenv("DIE_TOKEN")
     model_path = hf_hub_download(
         repo_id="gabar92/DIE",
         filename=args.die_model_path,