Update k8s-deployment.yaml

Author: cedricve

k8s-deployment.yaml

@@ -20,6 +20,7 @@ spec:
           resources:
             limits:
               nvidia.com/gpu: 1 # requesting a single GPU
+              nvidia.com/gpumem: 512 # Each vGPU contains 512m device memory
           env:
             - name: MODEL_NAME
               value: "yolov8n.pt"
@@ -69,4 +70,4 @@ spec:
             - name: FORWARDING_MEDIA
               value: "True"
             - name: REMOVE_AFTER_PROCESSED
-              value: "True"
+              value: "True"