Update main.py

main.py

@@ -6,11 +6,16 @@ from PIL import Image
 import cv2
 import base64
 import io
+import os
 
 app = FastAPI(title="OCR Base64 API")
 
-# Initialisation du lecteur EasyOCR (comme ton code)
-reader = easyocr.Reader(['en'], gpu=False, model_storage_directory='/tmp/easyocr')
+# Définir le répertoire pour EasyOCR
+model_storage_dir = '/code/.EasyOCR'
+os.makedirs(model_storage_dir, exist_ok=True)
+
+# Initialisation du lecteur EasyOCR avec le répertoire personnalisé
+reader = easyocr.Reader(['en'], gpu=False, model_storage_directory=model_storage_dir)
 
 # Modèle de requête
 class ImageRequest(BaseModel):
@@ -49,3 +54,7 @@ def extract_text(request: ImageRequest):
 
     except Exception as e:
         return {"error": str(e)}
+
+@app.get("/")
+def read_root():
+    return {"message": "OCR API is running!"}