update data loading

methods/utilities.py

@@ -44,17 +44,17 @@ def load_llama_llm(AUTH_TOKEN):
 def load_gdc_genes_mutations_hf(AUTH_TOKEN):
     dataset_id = 'uc-ctds/GDC-QAG-genes-mutations'
     filename = 'gdc_genes_mutations.json'
-    json_path = load_dataset(dataset_id, token=AUTH_TOKEN)
+    json_path = hf_hub_download(
+        repo_id=dataset_id,
+        filename=filename,
+        repo_type="dataset",
+        token=AUTH_TOKEN
+    )
+    # json_path = load_dataset(dataset_id, token=AUTH_TOKEN)
     with open(json_path, 'r') as f:
         gdc_genes_mutations = json.load(f)
     return gdc_genes_mutations
 
-# json_path = hf_hub_download(
-#     repo_id=dataset_id,
-#     filename=filename,
-#     repo_type="dataset",
-#     token=AUTH_TOKEN
-# )
 
 
 def load_intent_model_hf(AUTH_TOKEN):