Update my_model/object_detection.py

my_model/object_detection.py

@@ -239,9 +239,14 @@ def detect_and_draw_objects(image_path, model_type='yolov5', threshold=0.2, show
 
     detector = ObjectDetector()
     detector.load_model(model_type)
+    st.write("GGGGGGGGG")
     image = detector.process_image(image_path)
+    st.write("HHHHHHHHHHHH")
     detected_objects_string, detected_objects_list = detector.detect_objects(image, threshold=threshold)
+    st.write("YYYYYYYYYYYYYYYY")
     image_with_boxes = detector.draw_boxes(image, detected_objects_list, show_confidence=show_confidence)
+    st.write("KKKKKKKKKKKKKKK")
+    st.write(type(image_with_boxes))
     return image_with_boxes, detected_objects_string