Update main.py

main.py

@@ -1,9 +1,12 @@
+import os
+# Disable Numba caching to avoid the error
+os.environ["NUMBA_DISABLE_JIT"] = "1"
+
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.responses import StreamingResponse, FileResponse, RedirectResponse
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
 import io
-import os
 import sys
 from PIL import Image
 from rembg import remove
@@ -48,7 +51,8 @@ def health_check():
         "environment": {
             "python_version": sys.version,
             "directory": os.getcwd(),
-            "files": os.listdir()
+            "files": os.listdir(),
+            "numba_disabled": os.environ.get("NUMBA_DISABLE_JIT", "Not set")
         }
     }