Spaces:
Runtime error
Runtime error
update docstring
Browse files
summarizer/model_processors.py
CHANGED
|
@@ -342,7 +342,7 @@ class Summarizer(ModelProcessor):
|
|
| 342 |
|
| 343 |
class TransformerSummarizer(ModelProcessor):
|
| 344 |
"""
|
| 345 |
-
|
| 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 |
+
)
|