Spaces:
Runtime error
Runtime error
johnowhitaker
commited on
Commit
·
9be8b07
1
Parent(s):
fee9d2e
Corrected model names for light and dark models
Browse files
app.py
CHANGED
@@ -467,11 +467,11 @@ gan_new = gan_new.from_pretrained('johnowhitaker/orbgan_e1', latent_dim=256, ima
|
|
467 |
|
468 |
gan_light = Generator(latent_dim=256, image_size=256, attn_res_layers = [32])
|
469 |
gan_light.__class__ = GeneratorWithPyTorchModelHubMixin
|
470 |
-
gan_light = gan_light.from_pretrained('johnowhitaker/
|
471 |
|
472 |
gan_dark = Generator(latent_dim=256, image_size=256, attn_res_layers = [32])
|
473 |
gan_dark.__class__ = GeneratorWithPyTorchModelHubMixin
|
474 |
-
gan_dark = gan_dark.from_pretrained('johnowhitaker/
|
475 |
|
476 |
def gen_ims(n_rows, model='both'):
|
477 |
if model == "both":
|
|
|
467 |
|
468 |
gan_light = Generator(latent_dim=256, image_size=256, attn_res_layers = [32])
|
469 |
gan_light.__class__ = GeneratorWithPyTorchModelHubMixin
|
470 |
+
gan_light = gan_light.from_pretrained('johnowhitaker/orbgan_light', latent_dim=256, image_size=256, attn_res_layers = [32])
|
471 |
|
472 |
gan_dark = Generator(latent_dim=256, image_size=256, attn_res_layers = [32])
|
473 |
gan_dark.__class__ = GeneratorWithPyTorchModelHubMixin
|
474 |
+
gan_dark = gan_dark.from_pretrained('johnowhitaker/orbgan_dark', latent_dim=256, image_size=256, attn_res_layers = [32])
|
475 |
|
476 |
def gen_ims(n_rows, model='both'):
|
477 |
if model == "both":
|