Fix Label Name

app.py

@@ -27,8 +27,12 @@ def detect_objects(image):
 
     # Mostrar los objetos detectados
     detected_objects = []
+    id2label = model.config.id2label  # Diccionario ID -> Nombre
     for score, label, box in zip(scores, labels, boxes):
-        detected_objects.append(f"Objeto: {label}, Score: {score:.2f}, Box: {box.tolist()}")
+        label_name = id2label[int(label)]
+        detected_objects.append(
+            f"Objeto: {label_name}, Score: {score:.2f}, Box: {box.tolist()}"
+        )
 
     return "\n".join(detected_objects)