back to stable

app.py

@@ -41,8 +41,6 @@ def show_preds_image(image_path):
     outputs = model.predict(source=image_path)
     results = outputs[0].cpu().numpy()
     for i, det in enumerate(results.boxes.xyxy):
-        class_index = det[4].astype(int)
-        label = results.names[class_index]
         cv2.rectangle(
             image,
             (int(det[0]), int(det[1])),
@@ -51,21 +49,9 @@ def show_preds_image(image_path):
             thickness=2,
             lineType=cv2.LINE_AA
         )
-        cv2.putText(
-            image,
-            label,
-            (int(det[0]), int(det[1] - 10)),
-            cv2.FONT_HERSHEY_SIMPLEX,
-            0.5,
-            (255, 255, 255),
-            2,
-            lineType=cv2.LINE_AA
-        )
     return cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
 
 
-
-
 inputs_image = [
     gr.components.Image(type="filepath", label="Input Image"),
 ]
@@ -91,8 +77,6 @@ def show_preds_video(video_path):
             outputs = model.predict(source=frame)
             results = outputs[0].cpu().numpy()
             for i, det in enumerate(results.boxes.xyxy):
-                class_index = results.names[i]
-                label = class_index
                 cv2.rectangle(
                     frame_copy,
                     (int(det[0]), int(det[1])),
@@ -101,16 +85,6 @@ def show_preds_video(video_path):
                     thickness=2,
                     lineType=cv2.LINE_AA
                 )
-                cv2.putText(
-                    frame_copy,
-                    label,
-                    (int(det[0]), int(det[1] - 10)),
-                    cv2.FONT_HERSHEY_SIMPLEX,
-                    0.5,
-                    (255, 255, 255),
-                    2,
-                    lineType=cv2.LINE_AA
-                )
             yield cv2.cvtColor(frame_copy, cv2.COLOR_BGR2RGB)