Update app.py
Browse files
app.py
CHANGED
@@ -22,7 +22,7 @@ depth_model.to(device)
|
|
22 |
|
23 |
def process_image(image):
|
24 |
# Ensure image is in RGB format and resize
|
25 |
-
image =
|
26 |
|
27 |
# Perform segmentation
|
28 |
inputs = seg_feature_extractor(images=image, return_tensors="pt").to(device)
|
|
|
22 |
|
23 |
def process_image(image):
|
24 |
# Ensure image is in RGB format and resize
|
25 |
+
image = image.resize((512, 512))
|
26 |
|
27 |
# Perform segmentation
|
28 |
inputs = seg_feature_extractor(images=image, return_tensors="pt").to(device)
|