Update app.py

app.py

@@ -40,11 +40,13 @@ def fetch_model(url_or_path):
 def load_model(file_name: str, device: torch.device):
     #path = torch.hub.download_url_to_file('https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/'+f'{file_name}',
     #                       f'{file_name}')
-    base_url = "https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/"
-    network_url = base_url + f'{file_name}'
-    
-    local_path = '/content/'f'{file_name}'
-    with open(fetch_model(network_url), 'rb') as f:
+    #base_url = "https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/"
+    #network_url = base_url + f'{file_name}'
+    path = hf_hub_download('hysts/StyleGAN3', f'models/{file_name}')
+    #model_gen_n = torch.load(path, map_location=torch.device('cpu'))
+    #local_path = '/content/'f'{file_name}'
+    #with open(fetch_model(network_url), 'rb') as f:
+    with open(path, 'rb') as f:
         model = pickle.load(f)['G_ema']
     model.eval()
     model.to(device)