generate_face_embedding updated

unytics · May 13, 2024 · 7c5fb6d · 7c5fb6d
1 parent a144ad6
commit 7c5fb6d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bigfunctions/generate_face_embedding.yaml b/bigfunctions/generate_face_embedding.yaml
@@ -39,6 +39,7 @@ additional_install_commands_in_dockerfile: |
   RUN apt-get update -y && apt-get install -y libgl1-mesa-glx libglib2.0-0
   RUN python -c "import deepface.DeepFace; import deepface.detectors.DetectorWrapper; deepface.DeepFace.build_model('Facenet512'); deepface.detectors.DetectorWrapper.build_model('retinaface')"
 code: |
+  import tempfile
   import urllib.request
   from deepface import DeepFace
 
@@ -57,6 +58,6 @@ requirements: |
 quotas:
   max_rows_per_user_per_day: 1000
 cloud_run:
-  memory: 1024Mi
+  memory: 2048Mi
   concurrency: 1
   max_instances: 10