Added type checking

app.py

@@ -69,6 +69,10 @@ def generate_3d_model(depth, image_path, focallength_px):
     # Load the RGB image and convert to a NumPy array
     image = np.array(Image.open(image_path))
     
+    # Ensure depth is a NumPy array
+    if isinstance(depth, torch.Tensor):
+        depth = depth.cpu().numpy()
+    
     # Resize depth to match image dimensions if necessary
     if depth.shape != image.shape[:2]:
         depth = cv2.resize(depth, (image.shape[1], image.shape[0]), interpolation=cv2.INTER_LINEAR)
@@ -79,7 +83,7 @@ def generate_3d_model(depth, image_path, focallength_px):
     print(f"3D model generation - Image shape: {image.shape}")
 
     # Compute camera intrinsic parameters
-    fx = fy = focallength_px  # Assuming square pixels and fx = fy
+    fx = fy = float(focallength_px)  # Ensure focallength_px is a float
     cx, cy = width / 2, height / 2  # Principal point at the image center
 
     # Create a grid of (u, v) pixel coordinates