Update monocular_depth_estimator.py

monocular_depth_estimator.py

@@ -17,30 +17,24 @@ MODEL_FILE_URL = {
 
 class MonocularDepthEstimator:
     def __init__(self,
-        model_type="midas_v21_small_256",
-        model_weights_path="models/", 
-        optimize=False, 
-        side_by_side=False, 
-        height=None, 
-        square=False, 
-        grayscale=False):
-
-        # params
-        print("Initializing parameters...")
-        self.model_type = model_type
-        self.model_weights_path = model_weights_path
-        self.is_optimize = optimize
-        self.is_square = square
-        self.is_grayscale = grayscale
-        self.height = height
-        self.side_by_side = side_by_side
-        self.device = 'cuda'  # ZeroGPU will always use CUDA
-        
-        # Model will be loaded in make_prediction
-        self.model = None
-        self.transform = None
-        self.net_w = None
-        self.net_h = None
+    model_type="midas_v21_small_256",
+    model_weights_path="models/", 
+    optimize=False, 
+    side_by_side=False, 
+    height=None, 
+    square=False, 
+    grayscale=False):
+    
+    # Store parameters but don't initialize CUDA
+    self.model_type = model_type
+    self.model_weights_path = model_weights_path
+    self.is_optimize = optimize
+    self.is_square = square
+    self.is_grayscale = grayscale
+    self.height = height
+    self.side_by_side = side_by_side
+    self.model = None  # Model will be loaded in make_prediction
+    self.transform = None
 
         # Download model if not exists
         if not os.path.exists(model_weights_path+model_type+".pt"):