johnrobinsn commited on
Commit
29dbc5b
·
verified ·
1 Parent(s): 9bf12c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from pathlib import Path
7
  from depth_viewer import depthviewer2html
8
  import cv2
9
 
10
- feature_extractor = DPTFeatureExtractor.from_pretrained("Intel/dpt-large")
11
  model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
12
 
13
  def process_image(image_path):
 
7
  from depth_viewer import depthviewer2html
8
  import cv2
9
 
10
+ feature_extractor = DPTImageProcessor.from_pretrained("Intel/dpt-large")
11
  model = DPTForDepthEstimation.from_pretrained("Intel/dpt-large")
12
 
13
  def process_image(image_path):