update the arg classes in prediction#

Author: falibabaei

api/__init__.py

@@ -315,6 +315,7 @@ def main():
     args = cmd_parser.parse_args()
 
     main()
+    
 
     """
     python3 api/__init__.py  train --model yolov8n.yaml\

api/schemas.py

@@ -135,10 +135,13 @@ class Meta:
         },
         load_default=False,
     )
-    classes = fields.Field(
+    classes =  fields.List(
+        fields.Int(),
         metadata={
             "description": "Filter results by class, i.e. class=0, "
-            "or class=[0,2,3]"
+            "or class=[0,2,3]. Only detections belonging to the "
+            "specified classes will be returned. Useful for focusing"
+            " on relevant objects in multi-class detection tasks."
         },
         load_default=None,
     )