johnowhitaker commited on
Commit
9be8b07
·
1 Parent(s): fee9d2e

Corrected model names for light and dark models

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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/orbgan_ligt', 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_ligt', latent_dim=256, image_size=256, attn_res_layers = [32])
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":