caching

app.py

@@ -1,3 +1,10 @@
+import os
+PATH = '/data/'
+os.environ['TRANSFORMERS_CACHE'] = PATH
+os.environ['HF_HOME'] = PATH
+os.environ['HF_DATASETS_CACHE'] = PATH
+os.environ['TORCH_HOME'] = PATH
+
 import gradio as gr
 
 from spacy import glossary

utils.py

@@ -13,7 +13,6 @@ from interfaces.sentiment import build_huggingface_path as hf_sentiment_path
 from interfaces.emotion import build_huggingface_path as hf_emotion_path
 
 
-os.environ['TRANSFORMERS_CACHE'] = '/data/'
 HF_TOKEN = os.environ["hf_read"]
 
 models = [hf_manifesto_path(""), hf_sentiment_path(""), hf_emotion_path("")]