Spaces:
Runtime error
Runtime error
apolinario
commited on
Commit
·
22e38c7
1
Parent(s):
4a368e8
Remove capitalized...
Browse files
app.py
CHANGED
@@ -135,12 +135,12 @@ clip_model = CLIP()
|
|
135 |
|
136 |
#@markdown #**Model selection** 🎭
|
137 |
|
138 |
-
Models = ["
|
139 |
|
140 |
#@markdown ---
|
141 |
|
142 |
network_url = {
|
143 |
-
"
|
144 |
#"Imagenet512": "https://s3.eu-central-1.amazonaws.com/avg-projects/stylegan_xl/models/imagenet512.pkl",
|
145 |
#"Imagenet1024": "https://s3.eu-central-1.amazonaws.com/avg-projects/stylegan_xl/models/imagenet1024.pkl",
|
146 |
#"Pokemon": "https://s3.eu-central-1.amazonaws.com/avg-projects/stylegan_xl/models/pokemon256.pkl",
|
@@ -152,7 +152,7 @@ for Model in Models:
|
|
152 |
if not (path_exists(network_name)):
|
153 |
fetch_model(network_url[Model],network_name)
|
154 |
|
155 |
-
def load_current_model(current_model="
|
156 |
with dnnlib.util.open_url(current_model) as f:
|
157 |
G = legacy.load_network_pkl(f)['G_ema'].to(device)
|
158 |
|
|
|
135 |
|
136 |
#@markdown #**Model selection** 🎭
|
137 |
|
138 |
+
Models = ["imagenet256"]#, "Imagenet512", "Imagenet1024", "Pokemon", "FFHQ"]
|
139 |
|
140 |
#@markdown ---
|
141 |
|
142 |
network_url = {
|
143 |
+
"imagenet256":"https://s3.eu-central-1.amazonaws.com/avg-projects/stylegan_xl/models/imagenet256.pkl",
|
144 |
#"Imagenet512": "https://s3.eu-central-1.amazonaws.com/avg-projects/stylegan_xl/models/imagenet512.pkl",
|
145 |
#"Imagenet1024": "https://s3.eu-central-1.amazonaws.com/avg-projects/stylegan_xl/models/imagenet1024.pkl",
|
146 |
#"Pokemon": "https://s3.eu-central-1.amazonaws.com/avg-projects/stylegan_xl/models/pokemon256.pkl",
|
|
|
152 |
if not (path_exists(network_name)):
|
153 |
fetch_model(network_url[Model],network_name)
|
154 |
|
155 |
+
def load_current_model(current_model="imagenet256.pkl"):
|
156 |
with dnnlib.util.open_url(current_model) as f:
|
157 |
G = legacy.load_network_pkl(f)['G_ema'].to(device)
|
158 |
|