Update app.py

app.py

@@ -108,26 +108,29 @@ def process_image(image):
 
     return Image.fromarray(img)
 
-# Gradio Interface
-with gr.Blocks(title="Real-Time YOLOv5 Video & Image Detection") as demo:
-    gr.Markdown("# Real-Time YOLOv5 Object Detection")
-
+with gr.Blocks(title="Real-Time YOLOv5 Video & Image Object Detection") as demo:
+    gr.Markdown("""
+    # Real-Time YOLOv5 Object Detection
+    """, elem_id="title")
+    
     with gr.Tabs():
         with gr.TabItem("Video Detection"):
             with gr.Row():
-                video_input = gr.Video(label="Upload Video")
-                process_button = gr.Button("Process Video")
-            video_output = gr.Video(label="Processed Video")
+                video_input = gr.Video(label="Upload Video", interactive=True, elem_id="video-input")
+                process_button = gr.Button("Process Video", variant="primary", elem_id="video-process-btn")
+            video_output = gr.Video(label="Processed Video", elem_id="video-output")
             process_button.click(fn=process_video, inputs=video_input, outputs=video_output)
 
         with gr.TabItem("Image Detection"):
             with gr.Row():
-                image_input = gr.Image(type="pil", label="Upload Image", width=256, height=256)
-                image_output = gr.Image(label="Detected Objects", width=256, height=256)
+                image_input = gr.Image(type="pil", label="Upload Image", tool="editor", elem_id="image-input")
             with gr.Row():
-                clear_button = gr.Button("Clear")
-                submit_button = gr.Button("Submit")
+                image_output = gr.Image(label="Detected Objects", elem_id="image-output")
+            with gr.Row():
+                clear_button = gr.Button("Clear", variant="secondary", elem_id="clear-btn")
+                submit_button = gr.Button("Detect Objects", variant="primary", elem_id="submit-btn")
             clear_button.click(fn=lambda: None, inputs=None, outputs=image_output)
             submit_button.click(fn=process_image, inputs=image_input, outputs=image_output)
 
 demo.launch()
+