Fix: Model loading errors and cache permission issues

Dockerfile

@@ -8,7 +8,11 @@ RUN apt-get update && apt-get install -y \
     git \
     libgl1-mesa-glx \
     libglib2.0-0 \
+    wget \
     && rm -rf /var/lib/apt/lists/*
+    
+# Create cache directory with proper permissions
+RUN mkdir -p /.cache && chmod 777 /.cache
 
 # Copy requirements first for better caching
 COPY requirements.txt .

api.py

@@ -20,8 +20,19 @@ print("Loading models... This may take a moment.")
 # YOLOv8 model
 yolo_model = None
 try:
+    import os
     from ultralytics import YOLO
-    yolo_model = YOLO("yolov8n.pt")  # Using the nano model for faster inference
+    
+    # 모델 파일 경로
+    model_path = "yolov8n.pt"
+    
+    # 모델 파일이 없으면 직접 다운로드
+    if not os.path.exists(model_path):
+        print("Downloading YOLOv8 model...")
+        os.system("wget -q https://ultralytics.com/assets/yolov8n.pt -O yolov8n.pt")
+        print("YOLOv8 model downloaded successfully")
+    
+    yolo_model = YOLO(model_path)  # Using the nano model for faster inference
     print("YOLOv8 model loaded successfully")
 except Exception as e:
     print("Error loading YOLOv8 model:", e)