Generando mariposas

requeriments.txt

@@ -1,5 +1,2 @@
 git+https://github.com/huggingface/community-events.git@3fea10c5d5a50c69f509e34cd580fe9139905d04#egg=huggan
 transformers
-faiss-cpu
-paddlehub
-paddlepaddle

utils.py

@@ -1,9 +1,9 @@
 import numpy as np
 import torch
-from huggan.pytorch.lightweights_gan.lightweight_gan import LightweightGAN
+from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
 
-def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512",model_version=None):
-    gan = LightweightGAN.from_pretrained(model_name,version=model_version)
+def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):
+    gan = LightweightGAN.from_pretrained(model_name, version=model_version)
     gan.eval()
     return gan