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

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -0
src/streamlit_app.py CHANGED
@@ -15,6 +15,7 @@ load_dotenv()
15
  client = Together(api_key=os.getenv("TOGETHER_API_KEY", ""))
16
  import os
17
  os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
 
18
  # Set device
19
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
20
 
 
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
20
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
21