update from asus 106

backend/api/models.py

@@ -19,6 +19,7 @@ class TrainRequest(BaseModel):
     imgsz: Optional[int] = None
     device: Optional[Union[int, str]] = None
     batch_ids: Optional[list] = None
+    class_ids: Optional[list] = None
 
 
 @router.get("/api/hardware")
@@ -34,6 +35,7 @@ def start_training(project_id: int, request: TrainRequest) -> dict:
             project_id, request.epochs,
             {"batch": request.batch, "imgsz": request.imgsz, "device": request.device},
             batch_ids=request.batch_ids,
+            class_ids=request.class_ids,
         )
     except training.TrainingError as exc:
         raise HTTPException(400, str(exc))