Update app.py

app.py

@@ -444,7 +444,13 @@ def merge_loop():
     create_pod(model_name, USERNAME)
     # print(f"Total size: {human_readable_size(get_size(dir_path))}")
 
-    
+
+# Set the HF_DATASETS_CACHE environment variable
+os.environ['HF_DATASETS_CACHE'] = "/data/hfcache/"
+
+# Verify the environment variable is set
+print(os.environ['HF_DATASETS_CACHE'])
+
 # Install scrape-open-llm-leaderboard and mergekit
 command = ["git", "clone", "-q", "https://github.com/Weyaxi/scrape-open-llm-leaderboard"]
 subprocess.run(command, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)