ankban commited on
Commit
e022003
·
verified ·
1 Parent(s): e28dc9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -8,10 +8,10 @@ import openai
8
  from openai import OpenAI
9
 
10
  # Use writable cache for Hugging Face
11
- os.environ["XDG_CACHE_HOME"] = "/tmp/.cache"
12
  os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
13
- os.environ["HF_HOME"] = "/tmp/.cache/huggingface"
14
- os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache/huggingface"
 
15
 
16
  # Load Whisper and set OpenAI key
17
  model = WhisperModel("base", compute_type="int8") # Fastest CPU option
 
8
  from openai import OpenAI
9
 
10
  # Use writable cache for Hugging Face
 
11
  os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
12
+ os.environ["HF_HOME"] = "/data/hf"
13
+ os.environ["TRANSFORMERS_CACHE"] = "/data/hf"
14
+ os.environ["XDG_CACHE_HOME"] = "/data/hf"
15
 
16
  # Load Whisper and set OpenAI key
17
  model = WhisperModel("base", compute_type="int8") # Fastest CPU option