Update app.py

app.py

@@ -23,6 +23,7 @@ Related App: [StyleGAN-Human (Interpolation)](https://huggingface.co/spaces/hyst
 '''
 ARTICLE = '<center><img src="https://visitor-badge.glitch.me/badge?page_id=hysts.stylegan-human" alt="visitor badge"/></center>'
 
+TOKEN = os.environ['TOKEN']
 
 
 def parse_args() -> argparse.Namespace:
@@ -59,7 +60,8 @@ def generate_image(seed: int, truncation_psi: float, model: nn.Module,
 
 def load_model(file_name: str, device: torch.device) -> nn.Module:
     path = hf_hub_download('hysts/StyleGAN-Human',
-                           f'models/{file_name}')
+                           f'models/{file_name}',
+                           use_auth_token=TOKEN)
     with open(path, 'rb') as f:
         model = pickle.load(f)['G_ema']
     model.eval()