NavyaNayer commited on
Commit
4840860
·
verified ·
1 Parent(s): 85ec832

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -3
src/streamlit_app.py CHANGED
@@ -14,9 +14,9 @@ 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['SENTENCE_TRANSFORMERS_HOME'] = './.cache'
18
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
19
- os.environ["HF_HOME"] = "/tmp/huggingface"
20
  # Set device
21
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
22
 
 
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["HF_HOME"] = "/home/user/huggingface"
18
+
19
+
20
  # Set device
21
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
22