AAlexP commited on
Commit
1017190
·
verified ·
1 Parent(s): 78eb655

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,8 +4,10 @@ from fastapi import FastAPI, HTTPException, Request, Header
4
  from fastapi.middleware.cors import CORSMiddleware
5
  from PIL import Image
6
 
7
- # ✅ Fix cache path
8
- os.environ["TRANSFORMERS_CACHE"] = "/tmp"
 
 
9
 
10
  app = FastAPI(title="Skin Cancer Demo Inference")
11
 
 
4
  from fastapi.middleware.cors import CORSMiddleware
5
  from PIL import Image
6
 
7
+ # ✅ Put all HF caches in a writable place
8
+ os.environ["HF_HOME"] = "/tmp" # preferred going forward
9
+ os.environ["HUGGINGFACE_HUB_CACHE"] = "/tmp/hub" # optional, explicit
10
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers" # backward-compat
11
 
12
  app = FastAPI(title="Skin Cancer Demo Inference")
13