Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,6 @@ Related App: [StyleGAN-Human (Interpolation)](https://huggingface.co/spaces/hyst
|
|
23 |
'''
|
24 |
ARTICLE = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=hysts.stylegan-human" alt="visitor badge"/></center>'
|
25 |
|
26 |
-
TOKEN = os.environ['TOKEN']
|
27 |
|
28 |
|
29 |
def parse_args() -> argparse.Namespace:
|
@@ -60,8 +59,7 @@ def generate_image(seed: int, truncation_psi: float, model: nn.Module,
|
|
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:
|
66 |
model = pickle.load(f)['G_ema']
|
67 |
model.eval()
|
|
|
23 |
'''
|
24 |
ARTICLE = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=hysts.stylegan-human" alt="visitor badge"/></center>'
|
25 |
|
|
|
26 |
|
27 |
|
28 |
def parse_args() -> argparse.Namespace:
|
|
|
59 |
|
60 |
def load_model(file_name: str, device: torch.device) -> nn.Module:
|
61 |
path = hf_hub_download('hysts/StyleGAN-Human',
|
62 |
+
f'models/{file_name}')
|
|
|
63 |
with open(path, 'rb') as f:
|
64 |
model = pickle.load(f)['G_ema']
|
65 |
model.eval()
|