| # pipeline.yaml | |
| # The name of your pipeline | |
| name: "yolov5-inference" | |
| # The version of your pipeline | |
| version: "1.0" | |
| # The entry point script for your inference code | |
| entry_points: | |
| - script: inference.py | |
| # The description of your pipeline | |
| description: "YOLOv5 model for object detection inference." | |
| # The inputs for your pipeline | |
| inputs: | |
| - name: "image" | |
| type: "file" | |
| description: "Input image for object detection." | |
| # The outputs for your pipeline | |
| outputs: | |
| - name: "predictions" | |
| type: "table" | |
| description: "Object detection predictions." | |
| # The requirements for your pipeline | |
| requirements: | |
| - torch | |
| - torchvision | |
| - PIL | |