Refine Gradio UI and simplify README.md

README.md

@@ -8,12 +8,4 @@ sdk_version: 6.2.0
 app_file: app.py
 pinned: false
 license: cc-by-4.0
----
-
-# 🌾 HeadCount: Automated Wheat Head Counter
-
-Interactive demo for counting wheat heads in field images using semantic segmentation, using the model from [chmcbs/HeadCount](https://huggingface.co/chmcbs/HeadCount).
-
-- Designed for yield estimation, flowering time detection, and field maturity assessment
-- Best results with overhead imagery under diffuse lighting
-- May struggle with harsh lighting or heavily overlapping heads
+---

app.py

@@ -52,8 +52,11 @@ if os.path.exists("examples"):
 
 # Create Gradio interface
 with gr.Blocks(title="HeadCount") as demo:
-    gr.Markdown("# 🌾 HeadCount: Automated Wheat Head Counter")
-    gr.Markdown("Upload an image to automatically detect and count wheat heads.")
+    gr.Markdown("# 🌾 HeadCount")
+    gr.Markdown("### A semantic segmentation model for counting wheat heads in field images.")
+    gr.Markdown("- Designed for yield estimation, flowering time detection, and field maturity assessment")
+    gr.Markdown("- Best results with overhead imagery under diffuse lighting")
+    gr.Markdown("- Performance may vary with harsh lighting or heavily overlapping heads")
     
     with gr.Row():
         image_input = gr.Image(type="pil", label="Upload Image")
@@ -65,7 +68,6 @@ with gr.Blocks(title="HeadCount") as demo:
     with gr.Row():
         with gr.Column():
             if example_images:
-                gr.Markdown("### Example Images")
                 gr.Examples(examples=example_images, inputs=image_input)
         
         with gr.Column():
@@ -77,5 +79,7 @@ with gr.Blocks(title="HeadCount") as demo:
         outputs=[head_count_output, overlay_output]
     )
 
+    gr.Markdown("<div style='text-align: center'>Powered by <strong><a href='https://huggingface.co/chmcbs/HeadCount' target='_blank'>chmcbs/HeadCount</a></strong></div>")
+
 if __name__ == "__main__":
-    demo.launch(share=False, server_name="0.0.0.0", server_port=7860)
+    demo.launch(share=False, server_name="0.0.0.0", server_port=7860)