Fix: YOLOv8 model download permission issues using temp directory

api.py

@@ -23,14 +23,35 @@ try:
     import os
     from ultralytics import YOLO
     
-    # 모델 파일 경로
-    model_path = "yolov8n.pt"
+    # 모델 파일 경로 - 임시 디렉토리 사용
+    import tempfile
+    temp_dir = tempfile.gettempdir()
+    model_path = os.path.join(temp_dir, "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")
+        print(f"Downloading YOLOv8 model to {model_path}...")
+        try:
+            os.system(f"wget -q https://ultralytics.com/assets/yolov8n.pt -O {model_path}")
+            print("YOLOv8 model downloaded successfully")
+        except Exception as e:
+            print(f"Error downloading YOLOv8 model: {e}")
+            # 다운로드 실패 시 대체 URL 시도
+            try:
+                os.system(f"wget -q https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8n.pt -O {model_path}")
+                print("YOLOv8 model downloaded from alternative source")
+            except Exception as e2:
+                print(f"Error downloading from alternative source: {e2}")
+                # 마지막 대안으로 직접 모델 URL 사용
+                try:
+                    os.system(f"curl -L https://ultralytics.com/assets/yolov8n.pt --output {model_path}")
+                    print("YOLOv8 model downloaded using curl")
+                except Exception as e3:
+                    print(f"All download attempts failed: {e3}")
+    
+    # 환경 변수 설정 - 설정 파일 경로 지정
+    os.environ["YOLO_CONFIG_DIR"] = temp_dir
+    os.environ["MPLCONFIGDIR"] = temp_dir
     
     yolo_model = YOLO(model_path)  # Using the nano model for faster inference
     print("YOLOv8 model loaded successfully")