Update script.py

script.py

@@ -5,7 +5,14 @@ import json
 import yaml
 import subprocess
 
-HF_TOKEN = os.environ.get("HF_TOKEN")
+from huggingface_hub import login
+HF_TOKEN = os.getenv("HF_TOKEN")
+if HF_TOKEN:
+    login(token=HF_TOKEN)
+else:
+    raise ValueError("HF_TOKEN environment variable not found!")
+
+#HF_TOKEN = os.environ.get("HF_TOKEN")
 HF_DATASET = os.environ.get("DATA_PATH")