fix: cache directory

doppeltilde · Mar 1, 2024 · 8be2125 · 8be2125
1 parent 85ca9e0
commit 8be2125
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -17,7 +17,7 @@ services:
     ports:
       - "8000:8000"
     volumes:
-        - models:/app/models:rw
+      - models:/root/.cache/huggingface/hub:rw
     environment:
       - DEFAULT_MODEL_NAME
       - ACCESS_TOKEN

diff --git a/docker-compose.yml b/docker-compose.yml
@@ -6,7 +6,7 @@ services:
     ports:
       - "8000:8000"
     volumes:
-        - models:/app/models:rw
+      - models:/root/.cache/huggingface/hub:rw
     environment:
       - DEFAULT_MODEL_NAME
       - ACCESS_TOKEN
@@ -16,4 +16,4 @@ services:
     restart: unless-stopped
 
 volumes:
-  models:
+  models:
diff --git a/src/routes/api/video_classification.py b/src/routes/api/video_classification.py
@@ -1,7 +1,6 @@
 import cv2
 import os
 from fastapi import APIRouter, UploadFile, File, Query, Depends
-from dotenv import load_dotenv
 from typing import List
 import base64
 import asyncio
@@ -15,8 +14,6 @@
 
 executor = ThreadPoolExecutor()
 
-load_dotenv()
-
 
 def process_video(
     classifier,