Spaces:
Build error
Build error
File size: 1,294 Bytes
7e3e85d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
from model.base_model import SummModel
class MultiDocSummModel(SummModel):
is_multi_document = True
def __init__(
self,
trained_domain: str = None,
max_input_length: int = None,
max_output_length: int = None,
):
super(MultiDocSummModel, self).__init__(
trained_domain=trained_domain,
max_input_length=max_input_length,
max_output_length=max_output_length,
)
@classmethod
def assert_summ_input_type(cls, corpus, query):
if not all(
[
isinstance(ins, list) and all([isinstance(doc, str) for doc in ins])
for ins in corpus
]
):
raise TypeError(
"Multi-document summarization models summarize instances of multiple documents (`List[List[str]]`)."
)
if query is not None:
if not isinstance(query, list):
raise TypeError(
"Query-based single-document summarization requires query of `List[str]`."
)
if not all([isinstance(q, str) for q in query]):
raise TypeError(
"Query-based single-document summarization requires query of `List[str]`."
)
|