Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -28,8 +28,8 @@ import numpy as np
|
|
28 |
#pathlib.PosixPath = pathlib.WindowsPath
|
29 |
|
30 |
from huggingface_hub import hf_hub_download
|
31 |
-
ckpt = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-pruned.ckpt"
|
32 |
-
config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml"
|
33 |
|
34 |
device = "cuda:0"
|
35 |
model = load_model_from_config(config, ckpt, device=device, verbose=False)
|
|
|
28 |
#pathlib.PosixPath = pathlib.WindowsPath
|
29 |
|
30 |
from huggingface_hub import hf_hub_download
|
31 |
+
ckpt = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-pruned.ckpt")
|
32 |
+
config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml")
|
33 |
|
34 |
device = "cuda:0"
|
35 |
model = load_model_from_config(config, ckpt, device=device, verbose=False)
|