fix latest import

research_files/bench.py

@@ -33,8 +33,8 @@ if __name__ == '__main__':
         # "sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",  # Baseline Multilingual - sBERT
         # 'hiiamsid/sentence_similarity_spanish_es',          # Spanish similarity - sBERT
         # "jaimevera1107/all-MiniLM-L6-v2-similarity-es",     # Spanish similarity - sBERT
-        "google-bert/bert-base-multilingual-cased",                     # mBERT (google?)
-        "sentence-transformers/LaBSE",                      # LaBSE (google)
+        "google-bert/bert-base-multilingual-cased",                     # mBERT (google)
+        "sentence-transformers/LaBSE",                                  # LaBSE (google)
         "FacebookAI/xlm-roberta-base"                                  # XLM-R (facebook)
     ]:
         print("Evaluating Model (3 methods):", model)

research_files/inference/pipeline.py

@@ -8,7 +8,7 @@
 import numpy as np
 import torch
 from src.model import SegmentationNetwork
-from dataset import SegmentationTokenizer, SentenceSegmenter
+from src.dataset import SegmentationTokenizer, SentenceSegmenter
 
 
 # - # - # - # - # - # - # - # - # - # - # - # - # - # - # - #