Update CEFR_evaluator/model_base.py

CEFR_evaluator/model_base.py

@@ -5,7 +5,7 @@ from torch.utils.data import DataLoader
 import pytorch_lightning as pl
 from transformers import AutoTokenizer, AutoModel
 from sklearn.metrics import f1_score
-from util import mean_pooling, token_embeddings_filtering_padding, read_corpus, CEFRDataset, eval_multiclass
+from .util import mean_pooling, token_embeddings_filtering_padding, read_corpus, CEFRDataset, eval_multiclass
 
 class LevelEstimaterBase(pl.LightningModule):
     def __init__(self, corpus_path, test_corpus_path, pretrained_model, with_ib, attach_wlv,