Spaces:

danhergir
/

image_generator

Sleeping

danhergir commited on May 20

Commit

c51dff8

•

1 Parent(s): f7511b7

typo corrected

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
 def load_model(model_name='ceyda/butterfly_cropped_uniq1K_512', model_version=None):
-    gan = LightWeightGAN.from_pretrained(model_name, version=model_version)
     gan.eval()
     return gan

 from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
 def load_model(model_name='ceyda/butterfly_cropped_uniq1K_512', model_version=None):
+    gan = LightweightGAN.from_pretrained(model_name, version=model_version)
     gan.eval()
     return gan