Update 2.py

2.py

@@ -8,6 +8,20 @@ from tqdm import tqdm
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
+
+# Set the cache directory path
+cache_dir = '/app/cache'
+
+# Create the directory if it doesn't exist
+if not os.path.exists(cache_dir):
+    os.makedirs(cache_dir)
+
+# Set the environment variable
+os.environ['HF_HOME'] = cache_dir
+
+# Verify the environment variable is set
+print(f"HF_HOME is set to: {os.environ['HF_HOME']}")
+
 class Config: E, H, N, C, B, M, S, V, W, L, D = 512, 32, 1024, 256, 128, 20000, 2048, 1e5, 4000, 2e-4, .15
 
 class MyDataset(Dataset):