Update app.py

app.py

@@ -1,20 +1,20 @@
 import base64
 from gradio_client import Client
-from transformers import DPTFeatureExtractor, DPTForDepthEstimation
+from transformers import DPTImageProcessor, DPTForDepthEstimation
 import torch
 import numpy as np
 from PIL import Image
 import open3d as o3d
 from pathlib import Path
 import os
-import gradio as gr  # Add this import
+import gradio as gr
 
 # Initialize Gradio client for background removal
 client = Client("not-lain/background-removal")
 
-# Initialize DPT depth estimation model
-feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-large")
-model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
+# Initialize DPT depth estimation model (using the new ImageProcessor)
+feature_extractor = DPTImageProcessor.from_pretrained("Intel/dpt-large")
+model = DPTForDepthEstation.from_pretrained("Intel/dpt-large")
 
 def process_image(image_path, depth_map_path=None):
     image_path = Path(image_path)