RamAnanth1 commited on
Commit
0f4deee
·
1 Parent(s): c82be35

Update dataset/hf_dataset.py

Browse files
Files changed (1) hide show
  1. dataset/hf_dataset.py +3 -2
dataset/hf_dataset.py CHANGED
@@ -6,8 +6,9 @@ class HFDataset(ABC):
6
  """
7
  Create a dataset to save the transcripts from Youtube.
8
  """
9
- def __init__(self, name) -> None:
10
  self.name = name
 
11
  if name != "":
12
  self._init_dataset()
13
  else:
@@ -36,4 +37,4 @@ class HFDataset(ABC):
36
  pass
37
 
38
  def upload(self):
39
- self.dataset.push_to_hub(self.name)
 
6
  """
7
  Create a dataset to save the transcripts from Youtube.
8
  """
9
+ def __init__(self, name, token) -> None:
10
  self.name = name
11
+ self.token = token # HuggingFace Token
12
  if name != "":
13
  self._init_dataset()
14
  else:
 
37
  pass
38
 
39
  def upload(self):
40
+ self.dataset.push_to_hub(self.name, token = self.token)