Update app.py

app.py

@@ -1,27 +1,15 @@
 import gradio as gr
 from ultralytics import YOLO
 import cv2
-import os
 
-# Load YOLO model
 model = YOLO("best.pt")
 
 
-# -------------------------------
-# IMAGE DETECTION
-# -------------------------------
 def detect_image(image):
+    results = model(image, conf=0.5)
+    return results[0].plot()
 
-    results = model(image, conf=0.2)
 
-    result_img = results[0].plot()
-
-    return result_img
-
-
-# -------------------------------
-# VIDEO DETECTION
-# -------------------------------
 def detect_video(video):
 
     cap = cv2.VideoCapture(video)
@@ -34,23 +22,22 @@ def detect_video(video):
 
     out = cv2.VideoWriter(
         output_path,
-        cv2.VideoWriter_fourcc(*'mp4v'),
+        cv2.VideoWriter_fourcc(*"mp4v"),
         fps,
         (width, height)
     )
 
     while cap.isOpened():
-
         ret, frame = cap.read()
 
         if not ret:
             break
 
-        results = model(frame, conf=0.2)
+        results = model(frame, conf=0.5)
 
-        annotated_frame = results[0].plot()
+        annotated = results[0].plot()
 
-        out.write(annotated_frame)
+        out.write(annotated)
 
     cap.release()
     out.release()
@@ -58,84 +45,125 @@ def detect_video(video):
     return output_path
 
 
-# -------------------------------
-# UI
-# -------------------------------
-title = """
-# Kurshi's Project🙋🏻‍♀️🚗✨
-### Object Detection in Autonomous Vehicles
+# ---------------------------
+# CUSTOM CSS (your HTML style)
+# ---------------------------
+
+css = """
+
+body{
+background:linear-gradient(120deg,#f6f9fc,#e9f2ff);
+font-family:Segoe UI;
+}
+
+.main-container{
+max-width:950px;
+margin:auto;
+background:white;
+padding:35px;
+border-radius:20px;
+box-shadow:0px 10px 30px rgba(0,0,0,0.15);
+}
+
+.header{
+text-align:center;
+font-size:34px;
+font-weight:bold;
+color:#2c3e50;
+}
+
+.subheader{
+text-align:center;
+font-size:18px;
+margin-bottom:20px;
+color:#555;
+}
+
+.gr-button{
+background:#3498db !important;
+color:white !important;
+border-radius:10px !important;
+font-size:18px !important;
+padding:12px 30px !important;
+}
+
+.gr-button:hover{
+background:#217dbb !important;
+}
+
+footer{
+text-align:center;
+margin-top:20px;
+color:#777;
+font-size:14px;
+}
 """
 
-description = """
-Upload an **image or driving video** to detect objects such as:
 
-• Cars  
-• Trucks  
-• Buses  
-• Pedestrians  
-• Traffic Lights  
+# ---------------------------
+# UI
+# ---------------------------
 
-"""
+with gr.Blocks(css=css) as demo:
 
+    with gr.Column(elem_classes="main-container"):
 
-with gr.Blocks() as demo:
+        gr.Markdown(
+        """
+        <div class='header'>
+        Kurshi's Project 🙋🏻‍♀️✨
+        </div>
 
-    gr.Markdown(title)
-    gr.Markdown(description)
+        <div class='header'>
+        Object Detection in Autonomous Vehicles 🚗
+        </div>
 
-    with gr.Tabs():
+        <div class='subheader'>
+        YOLO Object Detection
+        </div>
+        """
+        )
 
-        # IMAGE TAB
-        with gr.Tab("📷 Image Detection"):
+        with gr.Tabs():
 
-            with gr.Row():
+            with gr.Tab("Upload Image"):
 
                 input_img = gr.Image(
                     type="numpy",
-                    label="Upload Road Image"
-                )
-
-                output_img = gr.Image(
-                    label="Detection Result"
+                    label="Upload Image"
                 )
 
-            detect_btn = gr.Button("🚀 Run Detection")
-
-            detect_btn.click(
-                fn=detect_image,
-                inputs=input_img,
-                outputs=output_img
-            )
+                detect_btn = gr.Button("Start Detection")
 
+                output_img = gr.Image(label="Detection Result")
 
-        # VIDEO TAB
-        with gr.Tab("🎥 Video Detection"):
+                detect_btn.click(
+                    detect_image,
+                    inputs=input_img,
+                    outputs=output_img
+                )
 
-            input_video = gr.Video(label="Upload Driving Video")
+            with gr.Tab("Upload Video"):
 
-            video_btn = gr.Button("🚀 Run Video Detection")
+                input_video = gr.Video(label="Upload Video")
 
-            output_video = gr.Video(label="Detection Result")
+                video_btn = gr.Button("Start Detection")
 
-            video_btn.click(
-                fn=detect_video,
-                inputs=input_video,
-                outputs=output_video
-            )
+                output_video = gr.Video(label="Detection Result")
 
+                video_btn.click(
+                    detect_video,
+                    inputs=input_video,
+                    outputs=output_video
+                )
 
-    gr.Markdown(
+        gr.Markdown(
         """
-        ---
-        ⚙️ Built using **YOLO26 + Computer Vision + Self Driving Datasets**
+        <footer>
+        YOLO | Self Driving Car Dataset | STET College
+        </footer>
         """
-    )
+        )
 
 
-# Launch app (compatible with HuggingFace Spaces)
-demo.launch(
-    server_name="0.0.0.0",
-    server_port=7860,
-    theme=gr.themes.Soft(),
-    share = True
-)
+demo.launch(server_name="0.0.0.0", server_port=7860, share = True)