Update app.py

app.py

@@ -25,7 +25,7 @@ def detect_salmon(image):
         filtered_detections = detection_data[detection_data[:, 4] >= conf_threshold]
 
         # Define class names (update based on your classes)
-        class_names = ["farmed", "wild"]
+        class_names = ["background", "farmed", "wild"]
 
         # Prepare boxes for NMS
         boxes = []