Update app.py
Browse files
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 =
|
| 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)
|