7 lines
268 B
Python
7 lines
268 B
Python
|
from transformers import pipeline
|
||
|
|
|
||
|
|
def process(input_text):
|
||
|
|
summarizer = pipeline("summarization", model="marianna13/flan-t5-base-summarization", max_length=200, min_length=10, device="mps")
|
||
|
|
output = summarizer(input_text)[0]['summary_text']
|
||
|
|
return output
|