Update app.py

app.py

@@ -81,10 +81,12 @@ def click_instance_segmentation(image, model_selection, predictor):
 
         st.write(f"Percentage of Inorganic Material in Nest: {percentage_inorganic_in_nest:.2f}%")
 
+        category_list = ['Plastic', 'Fishing_Net', 'Rope/Cloth', "Nest"]
+
         # Mostrar la cantidad de máscaras por categoría
         st.write("Mask Counts by Category:")
         for category, count in mask_counts.items():
-            st.write(f"Category {category}: {count} masks")
+            st.write(f"Category {category_list[category]}: {count} masks")
     
 
 def app():