feat: add handle dont draw if no tumor

app.py

@@ -34,13 +34,18 @@ async def inference(file: UploadFile):
     for r in results: 
         boxes = r.boxes
     for box in boxes: 
+        label = box.cls[0].item()  
+        label_name = model.names[label]  
+        
+        # Skip NoTumor class - don't draw bounding box
+        if label_name.lower() == "notumor":
+            continue
+        
         x1, y1, x2, y2 = box.xyxy[0]
         x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
     
         cv2.rectangle(img, (x1, y1), (x2, y2), (255, 0, 255), 2)
 
-        label = box.cls[0].item()  
-        label_name = model.names[label]  
         detections.append(label_name)
         (text_width, text_height), baseline = cv2.getTextSize(label_name, cv2.FONT_HERSHEY_SIMPLEX, 0.9, 2)
         cv2.rectangle(img, (x1, y1 - text_height - baseline - 5), (x1 + text_width, y1), (255, 0, 255), -1)
@@ -102,6 +107,13 @@ async def inference_volume(file: UploadFile):
         boxes = r.boxes
         
         for box in boxes: 
+            label = int(box.cls[0].item())
+            label_name = model.names[label]
+            
+            # Skip NoTumor class - don't draw bounding box or include in detections
+            if label_name.lower() == "notumor":
+                continue
+            
             x1, y1, x2, y2 = box.xyxy[0]
             x1, y1, x2, y2 = int(x1), int(y1), int(x2), int(y2)
             
@@ -112,8 +124,6 @@ async def inference_volume(file: UploadFile):
             volume_mm3 = calculate_sphere_volume(width, height)
             print("VOLUME")
             print(volume_mm3)
-            label = int(box.cls[0].item())
-            label_name = model.names[label]
             
             detections.append({
                 "class": label_name,