Spaces:
Running
Running
Update pbs_data.py
Browse files- pbs_data.py +2 -2
pbs_data.py
CHANGED
@@ -411,14 +411,14 @@ class PBSPublicDataAPIClient:
|
|
411 |
|
412 |
return processed
|
413 |
|
414 |
-
def save_data_to_hf(self, data, dataset_name="cmcmaster/rheumatology-biologics-dataset"):
|
415 |
processed_data = self.preprocess_data(data)
|
416 |
|
417 |
# Create a Dataset from the combinations
|
418 |
dataset = Dataset.from_list(processed_data['combinations'])
|
419 |
|
420 |
# Push the dataset to the Hugging Face Hub
|
421 |
-
dataset.push_to_hub(dataset_name)
|
422 |
|
423 |
print(f"Data saved to Hugging Face Hub: {dataset_name}")
|
424 |
|
|
|
411 |
|
412 |
return processed
|
413 |
|
414 |
+
def save_data_to_hf(self, data, hf_token, dataset_name="cmcmaster/rheumatology-biologics-dataset"):
|
415 |
processed_data = self.preprocess_data(data)
|
416 |
|
417 |
# Create a Dataset from the combinations
|
418 |
dataset = Dataset.from_list(processed_data['combinations'])
|
419 |
|
420 |
# Push the dataset to the Hugging Face Hub
|
421 |
+
dataset.push_to_hub(dataset_name, token=hf_token)
|
422 |
|
423 |
print(f"Data saved to Hugging Face Hub: {dataset_name}")
|
424 |
|