Create pipeline.yaml

pipeline.yaml

@@ -0,0 +1,32 @@
+# 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