Gladiator commited on
Commit
9e74753
1 Parent(s): 63f3553

update docstring

Browse files
Files changed (1) hide show
  1. summarizer/model_processors.py +2 -2
summarizer/model_processors.py CHANGED
@@ -342,7 +342,7 @@ class Summarizer(ModelProcessor):
342
 
343
  class TransformerSummarizer(ModelProcessor):
344
  """
345
- Newer style that has keywords for models and tokenizers, but allows the user to change the type.
346
  """
347
 
348
  MODEL_DICT = {
@@ -398,4 +398,4 @@ class TransformerSummarizer(ModelProcessor):
398
 
399
  super().__init__(
400
  None, model, tokenizer, hidden, reduce_option, sentence_handler, random_state, hidden_concat, gpu_id
401
- )
 
342
 
343
  class TransformerSummarizer(ModelProcessor):
344
  """
345
+ Another type of Summarizer class to choose keyword based model and tokenizer
346
  """
347
 
348
  MODEL_DICT = {
 
398
 
399
  super().__init__(
400
  None, model, tokenizer, hidden, reduce_option, sentence_handler, random_state, hidden_concat, gpu_id
401
+ )