KRISH09bha commited on
Commit
5332848
·
verified ·
1 Parent(s): 3722851

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -7,6 +7,9 @@ from PIL import Image
7
  import io
8
  import os
9
 
 
 
 
10
  app = FastAPI()
11
 
12
  MODEL_NAME = os.getenv("MODEL_NAME", "lmms-lab/LLaVA-OneVision-1.5-8B-Instruct")
 
7
  import io
8
  import os
9
 
10
+ os.environ["HF_HOME"] = "/app/.cache"
11
+ os.environ["HF_DATASETS_CACHE"] = "/app/.cache"
12
+ os.environ["TRANSFORMERS_CACHE"] = "/app/.cache"
13
  app = FastAPI()
14
 
15
  MODEL_NAME = os.getenv("MODEL_NAME", "lmms-lab/LLaVA-OneVision-1.5-8B-Instruct")