Update model.py

model.py

@@ -18,11 +18,8 @@ repo_dir = pathlib.Path(__file__).parent
 submodule_dir = repo_dir / 'ControlNet'
 sys.path.append(submodule_dir.as_posix())
 
-from annotator.canny import apply_canny
-from annotator.hed import apply_hed, nms
+
 from annotator.midas import apply_midas
-from annotator.mlsd import apply_mlsd
-from annotator.openpose import apply_openpose
 from annotator.uniformer import apply_uniformer
 from annotator.util import HWC3, resize_image
 
@@ -41,7 +38,7 @@ def download_all_controlnet_weights() -> None:
 class Model:
     def __init__(self,
                  base_model_id: str = 'runwayml/stable-diffusion-v1-5',
-                 task_name: str = 'canny'):
+                 task_name: str = 'depth'):
         self.device = torch.device(
             'cuda:0' if torch.cuda.is_available() else 'cpu')
         self.base_model_id = ''