Merge pull request #217 from bacetiner/master

Added the updated facade segmentation model
NHERI-SimCenter · Jul 18, 2024 · 631341c · 631341c
2 parents 10274e1 + 587dc90
commit 631341c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/brails/modules/FacadeParser/FacadeParser.py b/brails/modules/FacadeParser/FacadeParser.py
@@ -196,8 +196,9 @@ def install_default_model(model_path='tmp/models/facadeParser.pth'):
 
                 if not os.path.isfile(model_path):
                     print('Loading default facade parser model file to tmp/models folder...')
-                    torch.hub.download_url_to_file('https://zenodo.org/record/10448047/files/facadeParserv2.pth',
-                                                   model_path, progress=False)
+                    model_url = 'https://zenodo.org/record/10448047/files/facadeParserv2.pth'
+                    torch.hub.download_url_to_file(model_url, model_path, 
+                                                   progress=False)
                     print('Default facade parser model loaded')
                 else: 
                     print(f"Default facade parser model at {model_path} loaded")