Spaces:

danhtran2mind
/

autoencoder-grayscale2color-landscape

Running

danhtran2mind commited on 15 days ago

Commit

244926d

verified ·

1 Parent(s): 92888a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ WIDTH, HEIGHT = 512, 512
 load_model_path = "./ckpts/best_model.h5"
 if not os.path.exists(load_model_path):
     os.makedirs(os.path.dirname(load_model_path), exist_ok=True)
-    url = "https://huggingface.co/danhtran2mind/autoencoder-grayscale-to-color-landscape/resolve/main/ckpts/best_model.h5"
     print(f"Downloading model from {url}...")
     with requests.get(url, stream=True) as r:
         r.raise_for_status()

 load_model_path = "./ckpts/best_model.h5"
 if not os.path.exists(load_model_path):
     os.makedirs(os.path.dirname(load_model_path), exist_ok=True)
+    url = "https://huggingface.co/danhtran2mind/autoencoder-grayscale2color-landscape/resolve/main/ckpts/best_model.h5"
     print(f"Downloading model from {url}...")
     with requests.get(url, stream=True) as r:
         r.raise_for_status()