leonarb commited on
Commit
40acfc0
·
verified ·
1 Parent(s): 78640a7

Clean directory for cache, fix Gradio

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -2,7 +2,6 @@ import os
2
 
3
  # Set Hugging Face and Torch cache to a guaranteed-writable location
4
  cache_dir = "/tmp/huggingface_cache"
5
- os.environ["TRANSFORMERS_CACHE"] = cache_dir
6
  os.environ["HF_HOME"] = cache_dir
7
  os.environ["TORCH_HOME"] = cache_dir
8
 
@@ -124,4 +123,4 @@ demo = gr.Interface(
124
  )
125
 
126
  if __name__ == "__main__":
127
- demo.launch()
 
2
 
3
  # Set Hugging Face and Torch cache to a guaranteed-writable location
4
  cache_dir = "/tmp/huggingface_cache"
 
5
  os.environ["HF_HOME"] = cache_dir
6
  os.environ["TORCH_HOME"] = cache_dir
7
 
 
123
  )
124
 
125
  if __name__ == "__main__":
126
+ demo.launch(share = True)