Making dataset_name an env variable

run.py

@@ -5,9 +5,10 @@ from renumics.spotlight import dtypes as spotlight_dtypes
 import os
 
 if __name__ == "__main__":
-    dataset_name="ai-aerospace/ac-text-embedding-ada-002-ams-test"
+    # dataset_name="ai-aerospace/ac-text-embedding-ada-002-ams-test"
+    dataset_name=os.getenv("DATASET_NAME")
     # Load dataset using datasets.load_dataset()
-    dataset = datasets.load_dataset(dataset_name, split="train",data_dir=os.environ.get("HF_HOME"))
+    dataset = datasets.load_dataset(dataset_name, split="train")
     print("Dataset loaded using datasets.load_dataset().")
 
     df = dataset.to_pandas()