Update app.py

app.py

@@ -56,7 +56,7 @@ def detect_objects(frame):
                     
         return detections
 
-def display_detections(image, detections):
+def display_detections(input_image, output_image, detections):
     for det in detections:
         x1, y1, x2, y2, conf, detected_name = det
         color = colors.get(detected_name, (255, 255, 255))
@@ -78,7 +78,9 @@ def app():
             image = Image.open(uploaded_file)
             st.image(image, caption="Uploaded Image", use_column_width=True)
             detections = detect_objects(np.array(image))
-            display_detections(image, detections)
+            output_image = display_detections(np.array(image), np.array(image), detections)
+            st.image(output_image, caption="Output Image", use_column_width=True)
+
             
     elif input_type == "Video":
         st.write("Select an option to get the input video:")