Fix YOLOv5 serverless function (#4344)

cvat-ai · Feb 15, 2022 · 2f1e89c · 2f1e89c
1 parent f719648
commit 2f1e89c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/serverless/pytorch/ultralytics/yolov5/nuclio/function.yaml b/serverless/pytorch/ultralytics/yolov5/nuclio/function.yaml
@@ -96,12 +96,14 @@ spec:
   eventTimeout: 30s
   build:
     image: cvat/ultralytics-yolov5
-    baseImage: ultralytics/yolov5:latest
+    baseImage: ultralytics/yolov5:latest-cpu
 
     directives:
       preCopy:
         - kind: USER
           value: root
+        - kind: RUN
+          value: apt update && apt install --no-install-recommends -y libglib2.0-0
         - kind: WORKDIR
           value: /opt/nuclio