apolinario commited on
Commit
22e38c7
·
1 Parent(s): 4a368e8

Remove capitalized...

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -135,12 +135,12 @@ clip_model = CLIP()
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,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="Imagenet256.pkl"):
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