Update backend.py

backend.py

@@ -83,11 +83,11 @@ def building_model(building_version_dropdown, building_pth_dropdown, building_th
     return building_predictor
 
 # Model for LCZs
-def tree_model(lcz_version_dropdown, lcz_pth_dropdown, lcz_threshold, device="cpu"):
+def lcz_model(lcz_version_dropdown, lcz_pth_dropdown, lcz_threshold, device="cpu"):
     lcz_cfg = get_cfg()
     lcz_cfg.merge_from_file(get_version_cfg_yml("lcz_model_weights/lczs_cfg.yaml"))
     lcz_cfg.MODEL.DEVICE=device
-    lcz_cfg.MODEL.WEIGHTS = f"tree_model_weights/{tree_pth_dropdown}"
+    lcz_cfg.MODEL.WEIGHTS = f"tree_model_weights/{lcz_pth_dropdown}"
     lcz_cfg.MODEL.ROI_HEADS.NUM_CLASSES = 14  # TODO change this
     lcz_cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = lcz_threshold
     lcz_predictor = DefaultPredictor(lcz_cfg)