Update app.py

app.py

@@ -34,6 +34,7 @@ def fetch_model(url_or_path):
         return basename
     else:
         wget.download(url_or_path)
+        print(basename)
         return basename
            
 def load_model(file_name: str, device: torch.device):
@@ -43,8 +44,7 @@ def load_model(file_name: str, device: torch.device):
     network_url = base_url + f'{file_name}'
     
     local_path = '/content/'f'{file_name}'
-    print(local_path)
-    with open(local_path, 'rb') as f:
+    with open(fetch_model(network_url), 'rb') as f:
         model = pickle.load(f)['G_ema']
     model.eval()
     model.to(device)