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
|