Update app.py

app.py

@@ -54,7 +54,6 @@ def process_input(uploaded_file, youtube_link, image_url, sensitivity):
         return None, None, "Please provide an input using one of the methods.", ""
 
     try:
-        # Pass the slider value as the confidence threshold.
         results = model.predict(source=input_path, save=True, conf=sensitivity)
     except Exception as e:
         return None, None, f"Error running prediction: {e}", ""
@@ -84,14 +83,11 @@ with gr.Blocks(css="""
   object-fit: contain;
 }
 """) as demo:
-    # Header with scaled image (25% width) and title using absolute URL.
-    gr.HTML("<div style='text-align:center;'><img src='https://huggingface.co/spaces/tstone87/stance-detection/resolve/main/crowdresult.jpg' style='width:25%;'/></div>")
-    gr.Markdown("## Pose Detection with YOLO11-pose")
-    
-    # Create a two-column layout.
     with gr.Row():
-        # Left column: Input tabs and sensitivity slider.
+        # Left Column: Header image, title, input tabs and sensitivity slider.
         with gr.Column(scale=1):
+            gr.HTML("<div style='text-align:center;'><img src='https://huggingface.co/spaces/tstone87/stance-detection/resolve/main/crowdresult.jpg' style='width:25%;'/></div>")
+            gr.Markdown("## Pose Detection with YOLO11-pose")
             with gr.Tabs():
                 with gr.TabItem("Upload File"):
                     file_input = gr.File(label="Upload Image/Video")
@@ -99,9 +95,9 @@ with gr.Blocks(css="""
                     youtube_input = gr.Textbox(label="YouTube Link", placeholder="https://...")
                 with gr.TabItem("Image URL"):
                     image_url_input = gr.Textbox(label="Image URL", placeholder="https://...")
-            sensitivity_slider = gr.Slider(minimum=0.0, maximum=1.0, step=0.01, value=0.15,
+            sensitivity_slider = gr.Slider(minimum=0.0, maximum=1.0, step=0.01, value=0.5,
                                            label="Sensitivity (Confidence Threshold)")
-        # Right column: Display result.
+        # Right Column: Results display at the top.
         with gr.Column(scale=2):
             output_display = gr.Image(label="Annotated Output", elem_classes="result_img")
             output_file = gr.File(label="Download Annotated Output")