Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -59,7 +59,7 @@ def generate_image(seed: int, truncation_psi: float, model: nn.Module,
|
|
| 59 |
|
| 60 |
|
| 61 |
def load_model(file_name: str, device: torch.device) -> nn.Module:
|
| 62 |
-
path = hf_hub_download('
|
| 63 |
f'models/{file_name}',
|
| 64 |
use_auth_token=TOKEN)
|
| 65 |
with open(path, 'rb') as f:
|
|
|
|
| 59 |
|
| 60 |
|
| 61 |
def load_model(file_name: str, device: torch.device) -> nn.Module:
|
| 62 |
+
path = hf_hub_download('hysts/StyleGAN-Human',
|
| 63 |
f'models/{file_name}',
|
| 64 |
use_auth_token=TOKEN)
|
| 65 |
with open(path, 'rb') as f:
|