Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -453,8 +453,8 @@ def predict(
|
|
453 |
if isinstance(image, np.ndarray):
|
454 |
pil_image = Image.fromarray(image)
|
455 |
enhanced_image = ImageEnhance.Sharpness(pil_image).enhance(1)
|
456 |
-
|
457 |
-
image = np.array(
|
458 |
|
459 |
# ---------------------
|
460 |
# 1) Detect the drawer with YOLOWorld
|
|
|
453 |
if isinstance(image, np.ndarray):
|
454 |
pil_image = Image.fromarray(image)
|
455 |
enhanced_image = ImageEnhance.Sharpness(pil_image).enhance(1)
|
456 |
+
contrast_enhanced = ImageEnhance.Contrast(pil_image).enhance(1.5)
|
457 |
+
image = np.array(contrast_enhanced)
|
458 |
|
459 |
# ---------------------
|
460 |
# 1) Detect the drawer with YOLOWorld
|