bugfix
Browse files
app.py
CHANGED
@@ -112,7 +112,7 @@ def main():
|
|
112 |
for image_filename in example_images:
|
113 |
image_path = hf_hub_download(
|
114 |
repo_id="gabar92/DIE",
|
115 |
-
filename=f"args.example_image_path/{image_filename}",
|
116 |
use_auth_token=hf_token
|
117 |
)
|
118 |
example_image_list.append([Image.open(image_path)])
|
@@ -121,7 +121,7 @@ def main():
|
|
121 |
die_token = os.getenv("DIE_TOKEN")
|
122 |
model_path = hf_hub_download(
|
123 |
repo_id="gabar92/DIE",
|
124 |
-
filename=
|
125 |
use_auth_token=die_token
|
126 |
)
|
127 |
die_model = UNetDIEModel(args=args, model_path=model_path)
|
|
|
112 |
for image_filename in example_images:
|
113 |
image_path = hf_hub_download(
|
114 |
repo_id="gabar92/DIE",
|
115 |
+
filename=f"{args.example_image_path}/{image_filename}",
|
116 |
use_auth_token=hf_token
|
117 |
)
|
118 |
example_image_list.append([Image.open(image_path)])
|
|
|
121 |
die_token = os.getenv("DIE_TOKEN")
|
122 |
model_path = hf_hub_download(
|
123 |
repo_id="gabar92/DIE",
|
124 |
+
filename=args.die_model_path,
|
125 |
use_auth_token=die_token
|
126 |
)
|
127 |
die_model = UNetDIEModel(args=args, model_path=model_path)
|