Sarvamangalak commited on
Commit
f308121
·
verified ·
1 Parent(s): 3d3299c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -28,11 +28,11 @@ def classify_plate_color(plate_img):
28
  w = np.sum(white)
29
 
30
  if g > y and g > w:
31
- return "Electric Vehicle (Green Plate)"
32
  elif y > g and y > w:
33
- return "Commercial Vehicle (Yellow Plate)"
34
  else:
35
- return "Private Vehicle (White Plate)"
36
 
37
  # -------- Main Pipeline -------- #
38
  def process_image(img):
 
28
  w = np.sum(white)
29
 
30
  if g > y and g > w:
31
+ return " ELECTRIC Vehicle (Green Plate) "
32
  elif y > g and y > w:
33
+ return " COMMERCIAL Vehicle (Yellow Plate) "
34
  else:
35
+ return " PERSONAL Vehicle (White Plate) "
36
 
37
  # -------- Main Pipeline -------- #
38
  def process_image(img):