Update app.py

app.py

@@ -1,10 +1,16 @@
-import cv2
 from ultralytics import YOLO
-import gradio as gr
+import streamlit as st
 
 # Load the YOLOv8 model
 model = YOLO('best.pt')
 
+# video
+VIDEOS_DICT = {
+    'video_1': 'IMG_0559.mov',
+}
+
+# Your videos will start appearing inside streamlit webapp 'Choose a video'.
+
 # Define the Gradio interface
 def yolo_tracking(frame):
     # Run YOLOv8 tracking on the frame
@@ -18,8 +24,14 @@ def yolo_tracking(frame):
 
     return rgb_frame
 
-# Create the Gradio interface
-interface = gr.Interface(fn=yolo_tracking, inputs=gr.Video(), outputs=gr.Image(label="YOLOv8 Tracking"))
+# Setting page layout
+st.set_page_config(
+    page_title="Object Detection using YOLOv8",
+    page_icon="🤖",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
+# Main page heading
+st.title("Object Detection And Tracking using YOLOv8")
 
-# Launch the Gradio interface
-interface.launch()