Spaces:

joacorf33
/

butterflies

Runtime error

Joaquin Romero Flores commited on Mar 20, 2023

Commit

5a96139

•

1 Parent(s): 6f235ae

fixing spelling

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -11,7 +11,7 @@ from huggan.pytorch.lightweight_gan.leightweight_gan import LightweightGAN
 def load_model(model_name = "ceyda/butterfly_cropped_uniq1K_512", model_version=None):
     # GAN set-up
-    gan = LightweightGAN.from_pretrained(model_name, versio=model_version)
     # GAN Inference Evaluation
     gan.eval()
     return gan

 def load_model(model_name = "ceyda/butterfly_cropped_uniq1K_512", model_version=None):
     # GAN set-up
+    gan = LightweightGAN.from_pretrained(model_name, version=model_version)
     # GAN Inference Evaluation
     gan.eval()
     return gan