Update monocular_depth_estimator.py

monocular_depth_estimator.py

@@ -16,9 +16,9 @@ MODEL_FILE_URL = {
 }
 
 class MonocularDepthEstimator:
-    def __init__(self,
-        model_type="midas_v21_small_256",
-        model_weights_path="models/", 
+    def __init__(self, model_type="midas_v21_small_256", device="cpu"):
+        self.device = device
+        self.model, self.transform, *_ = load_model(self.device, f"models/{model_type}.pt", model_type)
         optimize=False, 
         side_by_side=False, 
         height=None,