Spaces:
Sleeping
Sleeping
File size: 730 Bytes
8fd2d1a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
# 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.
|