Update my_model/object_detection.py

my_model/object_detection.py

@@ -6,7 +6,7 @@ import numpy as np
 import cv2
 import os
 from my_model.utilities import get_path, show_image
-import streamlit as st
+
 
 
 class ObjectDetector:
@@ -250,7 +250,6 @@ def detect_and_draw_objects(image_path, model_type='yolov5', threshold=0.2, show
     image = detector.process_image(image_path)
     detected_objects_string, detected_objects_list = detector.detect_objects(image, threshold=threshold)
     image_with_boxes = detector.draw_boxes(image, detected_objects_list, show_confidence=show_confidence)
-    st.write(type(image_with_boxes))
     return image_with_boxes, detected_objects_string