cache dir

app.py

@@ -1,5 +1,6 @@
 import evaluate
 from evaluate.utils import launch_gradio_widget
 
-module = evaluate.load("my_perplexity", module_type="measurement")
+CACHE_DIR="/gfs/team/nlp/users/rcardena/tools/huggingface/evaluate"
+module = evaluate.load("my_perplexity", module_type="measurement",cache_dir=CACHE_DIR)
 launch_gradio_widget(module)

my_perplexity.py

@@ -107,6 +107,7 @@ class MyPerplexity(evaluate.Metric):
 
     def _compute(self, predictions, model_id, batch_size: int = 16, add_start_token: bool = True, device=None):
 
+        CACHE_DIR="/gfs/team/nlp/users/rcardena/tools/huggingface/evaluate"
         if device is not None:
             assert device in ["gpu", "cpu", "cuda"], "device should be either gpu or cpu."
             if device == "gpu":