Namra-Satva commited on
Commit
b65136b
·
verified ·
1 Parent(s): 32cfbac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -2,6 +2,18 @@ from fastapi import FastAPI, File, UploadFile
2
  from fastapi.responses import JSONResponse
3
  import shutil
4
  import os
 
 
 
 
 
 
 
 
 
 
 
 
5
  from model_utils import extract_invoice_data_from_image
6
 
7
  app = FastAPI()
 
2
  from fastapi.responses import JSONResponse
3
  import shutil
4
  import os
5
+
6
+ # ✅ Fix config/cache issues in Hugging Face Docker Space
7
+ os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
8
+ os.environ["YOLO_CONFIG_DIR"] = "/tmp/ultralytics"
9
+ os.environ["XDG_CACHE_HOME"] = "/tmp"
10
+ os.environ["FONTCONFIG_PATH"] = "/tmp"
11
+
12
+ # Create required directories if they don't exist
13
+ os.makedirs("/tmp/matplotlib", exist_ok=True)
14
+ os.makedirs("/tmp/ultralytics", exist_ok=True)
15
+ os.makedirs("/tmp/fontconfig", exist_ok=True)
16
+
17
  from model_utils import extract_invoice_data_from_image
18
 
19
  app = FastAPI()