bump

Author: jonafeucht

.env.example

@@ -1,10 +1,11 @@
+LOG_LEVEL=INFO
+
 DEFAULT_MODEL_NAME=onnx-community/nsfw_image_detection-ONNX
 ACCESS_TOKEN=
+DEFAULT_FACE_DETECTION_MODEL_URL=
 
 # False == Public Access
 # True == Access Only with API Key
 USE_API_KEY="False"
 API_KEY_HASH="<YOUR_GENERATED_KEY_HASH_HERE>"
-API_KEY_SALT="<YOUR_GENERATED_SALT_HERE>"
-
-LOG_LEVEL=INFO
+API_KEY_SALT="<YOUR_GENERATED_SALT_HERE>"

README.md

@@ -21,14 +21,8 @@ services:
       - ./cropped_faces:/app/cropped_faces:rw
       - ./models:/root/.cache/huggingface/hub:rw
       - ./mediapipe_models:/app/mediapipe_models:rw
-    environment:
-      - DEFAULT_MODEL_NAME
-      - ACCESS_TOKEN
-      - DEFAULT_FACE_DETECTION_MODEL_URL
-      - USE_API_KEY
-      - API_KEY_HASH
-      - API_KEY_SALT
-      - LOG_LEVEL
+    env_file:
+      - .env
     restart: unless-stopped
 ```
 
@@ -39,7 +33,7 @@ services:
 > You can find code examples in the [`examples`](./examples/) folder.
 
 ## Environment Variables
-- Create a `.env` file and set the preferred values.
+- Create a [`.env`](https://github.com/doppeltilde/vision_classification/blob/main/.env.example) file and set the preferred values.
 ```sh
 DEFAULT_MODEL_NAME=
 ACCESS_TOKEN=

docker-compose.dev.yml

@@ -9,12 +9,6 @@ services:
       - ./cropped_faces:/app/cropped_faces:rw
       - ./models:/root/.cache/huggingface/hub:rw
       - ./mediapipe_models:/app/mediapipe_models:rw
-    environment:
-      - DEFAULT_MODEL_NAME
-      - ACCESS_TOKEN
-      - DEFAULT_FACE_DETECTION_MODEL_URL
-      - USE_API_KEY
-      - API_KEY_HASH
-      - API_KEY_SALT
-      - LOG_LEVEL
+    env_file:
+      - .env
     restart: unless-stopped

docker-compose.swarm.yml

@@ -11,14 +11,8 @@ services:
       replicas: 3
       restart_policy:
         condition: on-failure
-    environment:
-      - DEFAULT_MODEL_NAME
-      - ACCESS_TOKEN
-      - DEFAULT_FACE_DETECTION_MODEL_URL
-      - USE_API_KEY
-      - API_KEY_HASH
-      - API_KEY_SALT
-      - LOG_LEVEL
+    env_file:
+      - .env
     restart: unless-stopped
     networks:
       - vision-network

docker-compose.yml

@@ -7,12 +7,6 @@ services:
       - ./cropped_faces:/app/cropped_faces:rw
       - ./models:/root/.cache/huggingface/hub:rw
       - ./mediapipe_models:/app/mediapipe_models:rw
-    environment:
-      - DEFAULT_MODEL_NAME
-      - ACCESS_TOKEN
-      - DEFAULT_FACE_DETECTION_MODEL_URL
-      - USE_API_KEY
-      - API_KEY_HASH
-      - API_KEY_SALT
-      - LOG_LEVEL
+    env_file:
+      - .env
     restart: unless-stopped

requirements.txt

@@ -2,7 +2,7 @@ fastapi[standard]==0.116.1
 Pillow==11.3.0
 filetype==1.2.0
 torch==2.7.1
-optimum[onnxruntime]==1.26.1
+optimum[onnxruntime]==1.27.0
 python-dotenv==1.1.1
 mediapipe
 numpy