Spaces:

danhergir
/

image_generator

Sleeping

danhergir commited on May 20

Commit

ce89ca9

•

1 Parent(s): 6649be3

Fixes

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -4,7 +4,7 @@ import os
 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, use_auth_token=os.environ["access_token"])
     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, use_auth_token=False)
     gan.eval()
     return gan