NavyaNayer commited on
Commit
9776b7d
·
verified ·
1 Parent(s): f827e93

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -0
src/streamlit_app.py CHANGED
@@ -14,6 +14,7 @@ load_dotenv()
14
  # Together AI Client with API key from environment variable
15
  client = Together(api_key=os.getenv("TOGETHER_API_KEY", ""))
16
  import os
 
17
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
18
  os.environ["HF_HOME"] = "/tmp/huggingface"
19
  # Set device
 
14
  # Together AI Client with API key from environment variable
15
  client = Together(api_key=os.getenv("TOGETHER_API_KEY", ""))
16
  import os
17
+ os.environ['SENTENCE_TRANSFORMERS_HOME'] = './.cache'
18
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
19
  os.environ["HF_HOME"] = "/tmp/huggingface"
20
  # Set device