Update app.py

app.py

@@ -60,8 +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}',
+    path = hf_hub_download('feng2022/Time-TravelRephotography',
+                           f'{file_name}',
                            use_auth_token=TOKEN)
     with open(path, 'rb') as f:
         model = pickle.load(f)['G_ema']