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:
```bash
pip install -r requirements.txt
```
2. Run the server locally:
```bash
python app.py
```
3. Deploy to [Hugging Face Spaces](https://huggingface.co/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.