diff --git a/serverless/pytorch/ultralytics/yolov5/nuclio/function.yaml b/serverless/pytorch/ultralytics/yolov5/nuclio/function.yaml
index 8fcb7435..4a4741c2 100644
--- 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