Update app.py

app.py

@@ -7,26 +7,34 @@ from PIL import Image
 from io import BytesIO
 import easyocr
 from ultralytics import YOLO
+import os
+
+# ✅ Force YOLO & EasyOCR cache dirs
+os.environ["EASYOCR_MODULE_PATH"] = "/app/.EasyOCR"
+os.environ["EASYOCR_CACHE_DIR"] = "/app/.EasyOCR"
+os.environ["YOLO_CONFIG_DIR"] = "/app/.yolo"
 
 app = FastAPI()
 
-# ✅ Root endpoint so "/" returns something
-@app.get("/")
-async def root():
-    return {"message": "YOLO + EasyOCR API is running!"}
+# ✅ Load EasyOCR from pre-downloaded cache
+reader = easyocr.Reader(['en'], model_storage_directory="/app/.EasyOCR")
 
-# Initialize OCR and YOLO
-reader = easyocr.Reader(['en'])
-model = YOLO("yolov8n.pt")
-print("✅ YOLO model loaded")
+# ✅ Load YOLO from pre-downloaded weights
+yolo_weights_path = "/root/.cache/torch/hub/checkpoints/yolov8n.pt"
+model = YOLO(yolo_weights_path)
+print("✅ YOLO model loaded from cache:", yolo_weights_path)
 
-# CORS settings
+# ✅ CORS settings
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"], allow_credentials=True,
     allow_methods=["*"], allow_headers=["*"],
 )
 
+@app.get("/")
+async def root():
+    return {"message": "YOLO Plate Detector API is running."}
+
 @app.post("/detect")
 async def detect_plate(file: UploadFile = File(...)):
     try: