Update utils/utils.py

utils/utils.py

@@ -6,6 +6,8 @@ from datasets import load_dataset
 
 # Dataset configuration
 
+SPACE_AUTHOR_NAME = os.getenv("SPACE_AUTHOR_NAME")
+
 DATASET_TYPE_GAIA      = "gaia"
 DATASET_TYPE_HLE       = "hle"
 
@@ -34,9 +36,7 @@ def get_dataset_from_file(dataset_type, level):
     return result
 
 def get_dataset(dataset_type, level):
-    space_id = os.environ.get("SPACE_ID", "bstraehle/gaia")
-    username = space_id.split("/")[0]
-    dataset_repo = f"{username}/validation"
+    dataset_repo = f"{SPACE_AUTHOR_NAME}/validation"
     dataset = load_dataset(dataset_repo, split="validation")
 
     df = dataset.to_pandas()