Update app.py

app.py

@@ -77,6 +77,9 @@ class ImageProcessor:
         for box in result.boxes:
             x1, y1, x2, y2 = [round(x) for x in box.xyxy[0].tolist()]
             class_id = box.cls[0].item()
+            # check if the class is tooth
+            if result.names[class_id].lower() == "tooth":
+                continue
             prob = round(box.conf[0].item(), 2)
             label = f"{result.names[class_id]}: {prob}"
             draw.rectangle([x1, y1, x2, y2], outline="red", width=2)