Spaces:
Sleeping
Sleeping
YOLO Segmentation Model Server (Gradio)
This subproject provides a Gradio web app and API endpoint for your YOLO segmentation model.
Quickstart
- Install dependencies:
pip install -r requirements.txt
- Run the server locally:
python app.py
- Deploy to Hugging Face Spaces:
- Push this directory to a new Space as a Gradio app.
Integration
- Replace the dummy
segment
function inapp.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.