gabar92 commited on
Commit
2c0f182
·
1 Parent(s): d07ad97
Files changed (1) hide show
  1. app.py +2 -2
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="args.die_model_path,
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)