Spaces:
Build error
Build error
from .single_doc import PegasusModel | |
class summarizer(PegasusModel): | |
def __init__(self, device="cpu"): | |
super(summarizer, self).__init__(device) | |
def show_capability(self): | |
print("Pegasus is the default singe-document summarization model.") | |
super(summarizer, self).show_capability() | |