codic commited on
Commit
c53c092
·
verified ·
1 Parent(s): 545379f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ client = Client("not-lain/background-removal")
14
 
15
  # Initialize DPT depth estimation model (using the new ImageProcessor)
16
  feature_extractor = DPTImageProcessor.from_pretrained("Intel/dpt-large")
17
- model = DPTForDepthEstation.from_pretrained("Intel/dpt-large")
18
 
19
  def process_image(image_path, depth_map_path=None):
20
  image_path = Path(image_path)
 
14
 
15
  # Initialize DPT depth estimation model (using the new ImageProcessor)
16
  feature_extractor = DPTImageProcessor.from_pretrained("Intel/dpt-large")
17
+ model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
18
 
19
  def process_image(image_path, depth_map_path=None):
20
  image_path = Path(image_path)