Update main.py

main.py

@@ -5,13 +5,22 @@ import tempfile
 import base64
 from typing import List
 
+os.environ.setdefault("HF_HUB_CACHE", "/tmp/hf_cache")
+os.environ.setdefault("HF_HOME", "/tmp/hf_home")
+os.environ.setdefault("HF_HUB_ENABLE_XET", "0")  # <-- disable xet backend
+os.environ.setdefault("MPLCONFIGDIR", "/tmp/mplconfig")
+
+os.makedirs("/tmp/hf_cache", exist_ok=True)
+os.makedirs("/tmp/hf_home", exist_ok=True)
+os.makedirs("/tmp/mplconfig", exist_ok=True)
+
 from fastapi import FastAPI, UploadFile, File, Form, HTTPException
 from fastapi.responses import HTMLResponse, JSONResponse
-from fastapi.staticfiles import StaticFiles
-from fastapi.middleware.cors import CORSMiddleware
+from fastapi.staticfiles import StaticFileseware
 
+from fastapi.middleware.cors import CORSMiddl
 import mmcv
-from model.run_inference import infer_images  # <- update this to accept model=...
+from model.run_inference import infer_images  
 
 logging.basicConfig(level=logging.INFO)