Update app.py

app.py

@@ -2,37 +2,20 @@ import streamlit as st
 import cv2
 import numpy as np
 from ultralytics import YOLO  
+from camera_input_live import camera_input_live
 
 # Load YOLO model (Download "yolov8n.pt" first if not available)
 model = YOLO("yolov8n.pt")  
 
 st.title("🎥 YOLO Object Detection Live Stream")
 
-# Open Webcam
-cap = cv2.VideoCapture(0)
-if not cap.isOpened():
-    st.error("Error: Could not access the webcam.")
-    st.stop()
 
-stframe = st.empty()  # Streamlit placeholder for video frames
+image = camera_input_live()
 
-while True:
-    ret, frame = cap.read()
-    if not ret:
-        st.error("Error: Failed to capture frame.")
-        break
+uploaded_image = PIL.Image.open(image)
+uploaded_image_cv = cv2.cvtColor(numpy.array(uploaded_image), cv2.COLOR_RGB2BGR)
+visualized_image = utils.predict_image(uploaded_image_cv, conf_threshold)
+results = model(frame)
+img_plot = result[0].plot()
 
-    # Convert frame to RGB (Streamlit expects RGB format)
-    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-
-    # Run YOLO object detection
-    results = model(frame)
-    img_pot = result[0].plot()
-
-  
-
-    # Show processed frame in Streamlit
-    stframe.image(img_pot, channels="RGB")
-
-# Release webcam
-cap.release()
+st.image(img_plot, channels = "BGR")