mhamilton723 commited on
Commit
b25dc1d
1 Parent(s): bb26736

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -24,6 +24,9 @@ if __name__ == "__main__":
24
  sample_videos_dir = "samples"
25
  else:
26
  os.environ['TORCH_HOME'] = '/tmp/.cache'
 
 
 
27
  os.environ['GRADIO_EXAMPLES_CACHE'] = '/tmp/gradio_cache'
28
  sample_videos_dir = "/tmp/samples"
29
 
 
24
  sample_videos_dir = "samples"
25
  else:
26
  os.environ['TORCH_HOME'] = '/tmp/.cache'
27
+ os.environ['HF_HOME'] = '/tmp/.cache'
28
+ os.environ['HF_DATASETS_CACHE'] = '/tmp/.cache'
29
+ os.environ['TRANSFORMERS_CACHE'] = '/tmp/.cache'
30
  os.environ['GRADIO_EXAMPLES_CACHE'] = '/tmp/gradio_cache'
31
  sample_videos_dir = "/tmp/samples"
32