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)
|