Update app.py

app.py

@@ -41,13 +41,10 @@ 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}'
-    path = hf_hub_download('hysts/StyleGAN3', f'models/{file_name}')
-    #model_gen_n = torch.load(path, map_location=torch.device('cpu'))
+    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:
-    with open(path, 'rb') as f:
+    with open(fetch_model(network_url), 'rb') as f:
         model = pickle.load(f)['G_ema']
     model.eval()
     model.to(device)