Merge branch 'main' of https://huggingface.co/spaces/joechou/test3 into main

dataset.py

@@ -8,7 +8,7 @@ import torch
 from torch.functional import Tensor
 from torch.nn.utils.rnn import pad_sequence
 from torch.utils.data.dataloader import DataLoader
-
+import helpers
 
 
 @dataclass
@@ -112,7 +112,7 @@ def from_processed(url: str, train=False):
 
 def get_collate_fn(text_vocab_size: int, max_seq_length: int):
     def collate_fn(examples: List[Example]) -> Batched:
-        from kobe.data.vocab import BOS_ID, EOS_ID
+        from vocab import BOS_ID, EOS_ID
 
         title_token_ids = pad_sequence(
             [