Jhol

app.py

@@ -186,8 +186,8 @@ async def detect_objects_yolo_world(
     image: UploadFile = File(...), # Revert back to File(...)
     profile: str = Form("casual"),
     extra_words: Optional[str] = Form(None),
-    confidence: float = Form(0.15, ge=0.0, le=1.0), # Lowered default confidence
-    iou: float = Form(0.40, ge=0.0, le=1.0) # Lowered default IoU (NMS threshold)
+    confidence: float = Form(0.2, ge=0.0, le=1.0), # Lowered default confidence
+    iou: float = Form(0.50, ge=0.0, le=1.0) # Lowered default IoU (NMS threshold)
 ):
     profile_lower = profile.lower()
     if profile_lower not in PREDEFINED_CLASSES:

detection_utils.py

@@ -49,10 +49,10 @@ PREDEFINED_CLASSES = {
         "knife", "spoon", "bowl", "banana", "apple", "sandwich", "orange",
         "broccoli", "carrot", "hot dog", "pizza", "donut", "cake", "chair",
         "couch", "potted plant", "bed", "dining table", "toilet", "tv",
-        "laptop", "mouse", "remote", "keyboard", "cell phone", "microwave",
-        "oven", "toaster", "sink", "refrigerator", "book", "clock", "vase",
-        "scissors", "teddy bear", "hair drier", "toothbrush", "tree", "flower", "park",
-        "computer", "monitor", "desk", "window", "door",
+        "laptop", "mouse", "tv remote","remote control", "keyboard", "cell phone", "microwave",
+        "oven", "toaster", "book", "clock",
+        "scissors", "teddy bear", "toothbrush", "tree", "flower", "park",
+        "computer", "desk", "window", "door",
         # Added Indian road/office-specific classes
         "auto rickshaw", "cycle rickshaw", "scooter", "tempo", "tractor", "e-rickshaw",
         "delivery van", "ambulance", "police car", "roadside stall", "food cart",
@@ -73,7 +73,7 @@ PREDEFINED_CLASSES = {
         "orange", "broccoli", "carrot", "hot dog", "pizza", "donut", "cake",
         "chair", "couch", "potted plant", "bed", "dining table", "toilet", "tv",
         "laptop", "mouse", "remote", "keyboard", "cell phone", "book", "clock",
-        "vase", "scissors", "teddy bear", "hair drier", "toothbrush", "red", "blue",
+         "scissors", "teddy bear", "hair drier", "toothbrush", "red", "blue",
         "green", "yellow", "orange", "purple", "pink", "black", "white", "gray",
         "brown", "circle", "square", "triangle", "rectangle", "star", "heart",
         "ball", "block", "toy", "doll", "crayon", "slide", "swing", "duck", "lion",