Spaces:
Runtime error
Runtime error
Commit
·
ba00f16
1
Parent(s):
b14c380
Update utils.py
Browse files
utils.py
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
import numpy as np
|
2 |
import torch
|
3 |
-
from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
|
4 |
-
|
5 |
|
6 |
## Cargamos el modelo desde el Hub de Hugging Face
|
7 |
def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):
|
|
|
1 |
import numpy as np
|
2 |
import torch
|
3 |
+
##from huggan.pytorch.lightweight_gan.lightweight_gan import LightweightGAN
|
4 |
+
from lightweight_gan import LightweightGAN
|
5 |
|
6 |
## Cargamos el modelo desde el Hub de Hugging Face
|
7 |
def carga_modelo(model_name="ceyda/butterfly_cropped_uniq1K_512", model_version=None):
|