Update src/app.py

src/app.py

@@ -8,12 +8,13 @@ import torch
 
 os.environ['HOME'] = '/tmp'
 
-os.environ['TRANSFORMERS_CACHE'] = '/tmp/hf_cache'
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/tr_cache'
 os.environ['HF_HOME'] = '/tmp/hf_cache'
+os.environ['HF_HUB_CACHE'] = '/tmp/hub_cache'
 
 os.environ['NLTK_DATA'] = '/tmp/nltk_data'
 
-cache_paths = ['/tmp/hf_cache', '/tmp/nltk_data']
+cache_paths = ['/tmp/hf_cache', '/tmp/nltk_data', '/tmp/tr_cache', '/tmp/hub_cache']
 for path in cache_paths:
     os.makedirs(path, exist_ok=True)