Update app.py

app.py

@@ -49,7 +49,7 @@ def yolo_inference(input_type, image, video, model_id, conf_threshold, iou_thres
     Returns:
         tuple: A tuple containing two elements:
             - PIL.Image.Image or None: The annotated image if `input_type` was "Image",
-                                       otherwise None.
+                                     otherwise None.
             - str or None: The path to the annotated video file if `input_type` was "Video",
                            otherwise None.
     """
@@ -191,7 +191,9 @@ def yolo_inference_for_examples(image, model_id, conf_threshold, iou_threshold,
     )
     return annotated_image
 
-with gr.Blocks() as app:
+theme = gr.themes.Ocean(primary_hue="blue", secondary_hue="pink")
+
+with gr.Blocks(theme=theme) as app:
     gr.Markdown("# Yolo13: Object Detection")
     gr.Markdown("Upload an image or video for inference using the latest YOLOv13 models.")
     gr.Markdown("📝 **Note:** Better-trained models will be deployed as they become available.")
@@ -229,14 +231,14 @@ with gr.Blocks() as app:
                 ],
                 value="yolov13n.pt",
             )
-            conf_threshold = gr.Slider(minimum=0, maximum=1, value=0.25, label="Confidence Threshold")
+            conf_threshold = gr.Slider(minimum=0, maximum=1, value=0.35, label="Confidence Threshold")
             iou_threshold = gr.Slider(minimum=0, maximum=1, value=0.45, label="IoU Threshold")
             max_detection = gr.Slider(minimum=1, maximum=300, step=1, value=300, label="Max Detection")
-            infer_button = gr.Button("Detect Objects")
+            infer_button = gr.Button("Detect Objects", variant="primary")
         with gr.Column():
             output_image = gr.Image(type="pil", show_label=False, show_share_button=False, visible=True)
             output_video = gr.Video(show_label=False, show_share_button=False, visible=False)
-            gr.DeepLinkButton()
+            gr.DeepLinkButton(variant="primary")
 
     input_type.change(
         fn=update_visibility,