Update app.py

app.py

@@ -28,6 +28,7 @@ model_paths = {
     "YOLOv8x Model": "yolov8x-doclaynet-epoch64-imgsz640-initiallr1e-4-finallr1e-5.pt",
     "YOLOv8m Model": "yolov8m-doclaynet.pt",
     "YOLOv8n Model": "yolov8n-doclaynet.pt",
+    "YOLOv8s Model": "yolov8s-doclaynet.pt",
     "DLA Model": "models/dla-model.pt"
 }
 
@@ -110,7 +111,7 @@ def process_image(image, model_choice):
 
 # Create the Gradio Interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Document Segmentation Demo (ZeroGPU)")
+    gr.Markdown("# Document Layout Segmentation Comparison (ZeroGPU)")
 
     with gr.Row():
         input_image = gr.Image(type="pil", label="Upload Image")