ammariii08 commited on
Commit
121681c
·
verified ·
1 Parent(s): 3a02ac9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- ei = ImageEnhance.Contrast(image).enhance(1.5)
457
- image = np.array(ei)
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