Update app.py

app.py

@@ -256,7 +256,7 @@ def detect_objects_video(model_name, video_input, threshold):
         labels = [model.config.id2label[x] for x in labels]
 
         for score, (xmin, ymin, xmax, ymax), label in zip(scores, boxes, labels):
-            if label == 'license-plates':
+            if "plate" in label.lower()::
                 crop = pil_img.crop((int(xmin), int(ymin), int(xmax), int(ymax)))
                 plate_type = classify_plate_color(crop)
 
@@ -290,19 +290,13 @@ def detect_objects_video(model_name, video_input, threshold):
 title = """<h1 id="title">Smart Vehicle Clssification (Image + Video)</h1>"""
 
 description = """
-Detect license plates using YOLOS or DETR.
-Supports:
-- Image URL
-- Image Upload
-- Webcam
-- Video Upload
-- Vehicle type classification by plate color
+Detect license plates using YOLOS or DETR and Vehicle classification.
+Supports:Image URL, Image Upload, Webcam, Video Upload 
 """
 
-models = [
-    "nickmuchi/yolos-small-finetuned-license-plate-detection"
-]
-
+#models = [
+#    "nickmuchi/yolos-small-finetuned-license-plate-detection"
+#]
 css = '''
 h1#title {
   text-align: center;