Update app.py

app.py

@@ -2,39 +2,24 @@ from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 import numpy as np
-import cv2
 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()
 
-# ✅ Load EasyOCR from pre-downloaded cache
-reader = easyocr.Reader(['en'], model_storage_directory="/app/.EasyOCR")
-
-# ✅ 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)
+# ✅ Use preloaded weights
+reader = easyocr.Reader(['en'], model_storage_directory='/app/.EasyOCR')
+model = YOLO("/app/yolov8n.pt")
+print("✅ YOLO model loaded from /app/yolov8n.pt")
 
-# ✅ 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: