summarize_job_post / summarizer.py
SamuelMiller's picture
Update summarizer.py
1b44afc
raw
history blame contribute delete
No virus
331 Bytes
# use bart in pytorch
summarizer = pipeline("summarization")
summarizer("An apple a day, keeps the doctor away", min_length=5, max_length=20)
# use t5 in tf
summarizer = pipeline("summarization", model="t5-base", tokenizer="t5-base", framework="tf")
summarizer("An apple a day, keeps the doctor away", min_length=5, max_length=20)