try fix running in hf.co

app.py

@@ -7,14 +7,13 @@ st.title('Numerai Example Script')
 # https://github.com/numerai/example-scripts/blob/master/example_model.py
 #
 
-IS_RUNNING_IN_HUGGING_FACE = False
 
 import pandas as pd
 from lightgbm import LGBMRegressor
 import gc
 import json
 from pathlib import Path
-
+import os
 from numerapi import NumerAPI
 from utils import (
     save_model,
@@ -28,6 +27,7 @@ from utils import (
     EXAMPLE_PREDS_COL
 )
 
+IS_RUNNING_IN_HUGGING_FACE = os.environ.get('HF_ENDPOINT') is not None
 
 napi = NumerAPI()
 current_round = napi.get_current_round()
@@ -38,15 +38,19 @@ current_round = napi.get_current_round()
 
 Path("./v4").mkdir(parents=False, exist_ok=True)
 
-
 @st.cache
-def download_dataset():
-    print('download_dataset')
-
+def get_dataset_path():
     if IS_RUNNING_IN_HUGGING_FACE:
         from datasets import load_dataset_builder
         ds_builder = load_dataset_builder("Numerati/numerai-datasets")
+        return ds_builder.cache_dir
     else:
+        return "./v4"
+
+@st.cache
+def download_dataset():
+    print('download_dataset')
+    if IS_RUNNING_IN_HUGGING_FACE:
         napi.download_dataset("v4/train.parquet")
         napi.download_dataset("v4/validation.parquet")
         napi.download_dataset("v4/validation_example_preds.parquet")