Update app.py

app.py

@@ -34,19 +34,14 @@ def detect_objects(image, score_threshold):
             if score >= score_threshold:  # Only draw if score is above threshold
                 no_detection = False
                 box = [round(i, 2) for i in box.tolist()]
-                
-                if label.item() == 0:
-                    label_name = "Pneumonia"
-                else:
-                    label_name = "Other"
-                    
+                label_name = "Pneumonia" if label.item() == 0 else "Other"   
                 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 detections, set label as 'Other'
     if no_detection:
-        labels_output.append("Other")
+        labels_output.append("No Detection")
     
     return image, "\n".join(labels_output)