Update app.py

app.py

@@ -7,6 +7,16 @@ from ultralytics import ASSETS, YOLO
 model_path = 'new_data_improved_object_detector.pt'
 model = YOLO(model_path)
 
+def Get_logo_xywh(model_result_input):
+    model_result = model_result_input[0]
+    xywh = model_result.boxes.xywh.cpu().tolist()
+    clss = model_result.boxes.cls.cpu().tolist()
+    # names = model_result_input[0].names
+    confidence = model_result.boxes.conf.cpu().tolist()
+    xyxy = model_result.boxes.cpu().xyxy.tolist()
+
+    return xywh, clss, confidence, xyxy
+
 def predict_image(img, conf_threshold, iou_threshold):
     """Predicts and plots labeled objects in an image using YOLOv8 model with adjustable confidence and IOU thresholds."""
     # Convert the input image to grayscale
@@ -22,33 +32,15 @@ def predict_image(img, conf_threshold, iou_threshold):
     )
 
     im_arrays = []
-    all_model_result=[]
-    all_xywh = []
-    all_clss = []
-    all_names = []
-    all_confidence = []
-    all_xyxy =[]
 
     for r in results:
-        model_result =results[0]
-        xywh = r.boxes.xywh.cpu().tolist()
-        clss = r.boxes.cls.cpu().tolist()
-        names = [r.names[cls] for cls in clss]  # Convert class indices to names
-        confidence = r.boxes.conf.cpu().tolist()
-        xyxy = r.boxes.cpu().xyxy.tolist()
-
         im_array = r.plot()
         im = Image.fromarray(im_array[..., ::-1])
         im_arrays.append(im)
-        
-        all_model_result.extend(model_result)
-        all_xywh.extend(xywh)
-        all_clss.extend(clss)
-        all_names.extend(names)
-        all_confidence.extend(confidence)
-        all_xyxy.extend(xyxy)
 
-    return im_arrays, model_result,all_xywh, all_clss, all_names, all_confidence,xyxy
+    xywh, clss,  confidence, xyxy = Get_logo_xywh(results)
+
+    return im_arrays, xywh, clss, confidence, xyxy
 
 iface = gr.Interface(
     fn=predict_image,
@@ -59,10 +51,8 @@ iface = gr.Interface(
     ],
     outputs=[
         gr.Gallery(label="Result Images"),
-        gr.JSON(label="Detection Bounding Boxes (model_result)"),
         gr.JSON(label="Detection Bounding Boxes (xywh)"),
         gr.JSON(label="Detection Class Indices"),
-        gr.JSON(label="Detection Class Names"),
         gr.JSON(label="Detection Confidence Scores"),
         gr.JSON(label="Detection Bounding Boxes (xyxy)")
     ],