Gladiator commited on
Commit
8d3b04a
1 Parent(s): 5af0584

fixed import

Browse files
extractive_summarizer/model_processors.py CHANGED
@@ -12,9 +12,9 @@ from transformers import (AlbertModel, AlbertTokenizer, BartModel,
12
  TransfoXLModel, TransfoXLTokenizer, XLMModel,
13
  XLMTokenizer, XLNetModel, XLNetTokenizer)
14
 
15
- from summarizer.bert_parent import BertParent
16
- from summarizer.cluster_features import ClusterFeatures
17
- from summarizer.sentence_handler import SentenceHandler
18
 
19
 
20
  class ModelProcessor(object):
 
12
  TransfoXLModel, TransfoXLTokenizer, XLMModel,
13
  XLMTokenizer, XLNetModel, XLNetTokenizer)
14
 
15
+ from extractive_summarizer.bert_parent import BertParent
16
+ from extractive_summarizer.cluster_features import ClusterFeatures
17
+ from extractive_summarizer.sentence_handler import SentenceHandler
18
 
19
 
20
  class ModelProcessor(object):