Update save_data.py

save_data.py

@@ -4,11 +4,10 @@ import json
 import requests
 import re as r
 from urllib.request import urlopen
-# from huggingface_hub import Repository
-from datasets import load_dataset
+from huggingface_hub import Repository
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
-DATASET_NAME = load_dataset("SRDdev/QABERT-small")
+DATASET_NAME = "bert_based_qna_dataset"
 DATASET_REPO_URL = f"https://huggingface.co/datasets/pragnakalp/{DATASET_NAME}"
 DATA_FILENAME = "qna_logs.csv"
 DATA_FILE = os.path.join("qna_logs", DATA_FILENAME)