Aayan873 commited on
Commit
9889ee0
·
verified ·
1 Parent(s): 046ad60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -10
app.py CHANGED
@@ -1,17 +1,19 @@
1
  # Streamlit App for Multimodal RAG System - Deployment Optimized
2
  import os, pathlib
3
- cache_root = pathlib.Path(os.getcwd())/".cache"
4
- (cache_root/"huggingface").mkdir(parents=True, exist_ok=True)
5
- (cache_root/"torch").mkdir(parents=True, exist_ok=True)
6
- os.environ.setdefault("HF_HOME", str(cache_root/"huggingface"))
7
- os.environ.setdefault("TRANSFORMERS_CACHE", str(cache_root/"huggingface"))
8
- os.environ.setdefault("HUGGINGFACE_HUB_CACHE", str(cache_root/"huggingface"))
9
- os.environ.setdefault("HF_DATASETS_CACHE", str(cache_root/"huggingface"))
10
- os.environ.setdefault("SENTENCE_TRANSFORMERS_HOME", str(cache_root/"huggingface"))
11
- os.environ.setdefault("XDG_CACHE_HOME", str(cache_root))
12
- os.environ.setdefault("TORCH_HOME", str(cache_root/"torch"))
 
13
  os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
14
 
 
15
  import streamlit as st
16
  import os
17
  import tempfile
 
1
  # Streamlit App for Multimodal RAG System - Deployment Optimized
2
  import os, pathlib
3
+ cache_root = pathlib.Path("/tmp")
4
+ (os.makedirs := None) # placeholder to satisfy linters if needed
5
+ (pathlib.Path("/tmp/hf")).mkdir(parents=True, exist_ok=True)
6
+ (pathlib.Path("/tmp/torch")).mkdir(parents=True, exist_ok=True)
7
+ os.environ.setdefault("HF_HOME", "/tmp/hf")
8
+ os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/hf")
9
+ os.environ.setdefault("HUGGINGFACE_HUB_CACHE", "/tmp/hf")
10
+ os.environ.setdefault("HF_DATASETS_CACHE", "/tmp/hf")
11
+ os.environ.setdefault("SENTENCE_TRANSFORMERS_HOME", "/tmp/hf")
12
+ os.environ.setdefault("XDG_CACHE_HOME", "/tmp")
13
+ os.environ.setdefault("TORCH_HOME", "/tmp/torch")
14
  os.environ.setdefault("TOKENIZERS_PARALLELISM", "false")
15
 
16
+
17
  import streamlit as st
18
  import os
19
  import tempfile