Update app.py

app.py

@@ -32,10 +32,14 @@ def detect_objects(image, score_threshold):
         for score, label, box in zip(scores, labels, boxes):
             if score >= score_threshold:  # Only draw if score is above threshold
                 box = [round(i, 2) for i in box.tolist()]
-                label_name = "Pneumonia" if label.item() == 0 else "Other"
+                label_name = "Pneumonia" if label.item() == 0 else "No detection"
                 draw.rectangle(box, outline="red", width=3)
                 draw.text((box[0], box[1]), f"{label_name}: {round(score.item(), 3)}", fill="red")
                 labels_output.append(f"{label_name}: {round(score.item(), 3)}")
+
+    # If no objects detected, append "No detection"
+    if not labels_output:
+        labels_output.append("No detection")
     
     return image, "\n".join(labels_output)