Hugging Face's logo

Spaces:
kamwoh
/
dreamcreature
Running

App Files Community
2
kamwoh commited on
Commit
0a52f97
•
1 Parent(s): f44c040

fixed

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def prepare_pipeline(model_name):
62
  checkpoint_name = {
63
  'dreamcreature-sd1.5-cub200': 'checkpoint-74900',
64
  'dreamcreature-sd1.5-dog': 'checkpoint-150000'
65
- }
66
 
67
  repo_url = f"https://huggingface.co/kamwoh/{model_name}/resolve/main"
68
  file_url = repo_url + f"/{checkpoint_name}/pytorch_model.bin"
 
62
  checkpoint_name = {
63
  'dreamcreature-sd1.5-cub200': 'checkpoint-74900',
64
  'dreamcreature-sd1.5-dog': 'checkpoint-150000'
65
+ }[model_name]
66
 
67
  repo_url = f"https://huggingface.co/kamwoh/{model_name}/resolve/main"
68
  file_url = repo_url + f"/{checkpoint_name}/pytorch_model.bin"