clusterflux / README.md
rayh's picture
Deploy latest YOLO model and app (version 20250422.7)
8fd2d1a
|
raw
history blame
730 Bytes

YOLO Segmentation Model Server (Gradio)

This subproject provides a Gradio web app and API endpoint for your YOLO segmentation model.

Quickstart

  1. Install dependencies:
pip install -r requirements.txt
  1. Run the server locally:
python app.py
  1. Deploy to Hugging Face Spaces:
    • Push this directory to a new Space as a Gradio app.

Integration

  • Replace the dummy segment function in app.py with your YOLO model inference code.
  • The API will accept an image and return a segmentation mask (as an image).

Notes

  • Add any additional dependencies to requirements.txt as needed.
  • For Spaces, ensure your model weights are included or downloadable.