Refactor object detection output to return a formatted string of detected objects

app.py

@@ -35,27 +35,35 @@ def detect_objects(image):
     for score, label, box in zip(scores, labels, boxes):
         detected_objects.append(f"Objeto: {label}: {model.config.id2label[label.item()]}, Score: {score:.2f}, Box: {box.tolist()}")
         
-    ##Plotear los resultados
-    plt.figure(figsize=(10, 10))
-    plt.imshow(image)
-    ax = plt.gca()
-    for score, label, box in zip(scores, labels, boxes):
-        x_min, y_min, x_max, y_max = box
-        rect = Rectangle((x_min, y_min), x_max - x_min, y_max - y_min,
-                         fill=False, color='red', linewidth=2)
-        ax.add_patch(rect)
-        ax.text(x_min, y_min, f"{model.config.id2label[label.item()]}: {score:.2f}", 
-                fontsize=12, color='red', verticalalignment='top')
-    plt.axis('off')
-    plt.tight_layout()
+    # # Visualizar los resultados de la detección
+    # plt.figure(figsize=(10, 10))
+    # plt.imshow(image)
+    # ax = plt.gca()
+    # for score, label, box in zip(scores, labels, boxes):
+    #     x_min, y_min, x_max, y_max = box
+    #     rect = Rectangle((x_min, y_min), x_max - x_min, y_max - y_min,
+    #                      fill=False, color='red', linewidth=2)
+    #     ax.add_patch(rect)
+    #     ax.text(x_min, y_min, f"{model.config.id2label[label.item()]} | Total score: {score:.2f}", 
+    #             fontsize=12, color='red', verticalalignment='top')
+    # plt.axis('off')
+    # plt.tight_layout()
+    # fig = plt.gcf()
+    # buf = io.BytesIO()
+    # fig.savefig(buf, format='png', bbox_inches='tight')
+    # buf.seek(0)
+    # plt.close()
+    
+    #Retornar la imagen con las detecciones
+    # return Image.open(buf)
     
-    #return image con los objetos detectados
-    fig = plt.gcf()
-    buf = io.BytesIO()
-    fig.savefig(buf, format='png', bbox_inches='tight')
-    buf.seek(0)
-    plt.close()
-    return Image.open(buf)
+     # Mostrar los objetos detectados
+
+    detected_objects = []
+    for score, label, box in zip(scores, labels, boxes):
+        detected_objects.append(f"Objeto: {label}, Score: {score:.2f}, Box: {box.tolist()}")
+        
+    return "\n".join(detected_objects)
 
 # Crear la interfaz Gradio
 def create_interface():