Update app.py

app.py

@@ -3,10 +3,10 @@ import gradio as gr
 from ultralytics import YOLO
 from PIL import Image
 
-# Load YOLOv8 model (make sure best.pt is in the same folder as this script)
+# Load YOLOv8 model 
 model = YOLO('best.pt')
 
-# Folder with example test images (must be inside your repo for HF Spaces)
+# Folder with example test images 
 test_images_folder = 'test_images'
 test_images = sorted(os.listdir(test_images_folder))
 
@@ -35,7 +35,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         uploaded_image = gr.Image(label="Upload your image (optional)", type="filepath")
         selected_image = gr.Dropdown(choices=test_images, label="Or choose a test image")
 
-    gr.Markdown("### 📸 Prediction Output")
+    gr.Markdown("###  Prediction Output")
     output_image = gr.Image(label="Predicted Output")
 
     gr.Button("Run prediction").click(