|
"""Build a Hugging Face Summarize Library""" |
|
|
|
from transformers import pipeline |
|
import wikipedia |
|
|
|
|
|
def summarize(text): |
|
"""Summarize text""" |
|
summarizer = pipeline("summarization", model="t5-small") |
|
result = summarizer(text, max_length=180) |
|
print("Summarization process complete!") |
|
print("=" * 80) |
|
return result[0]["summary_text"] |
|
|
|
|
|
|
|
def search_wiki(search_term): |
|
"""Search wikipedia pages""" |
|
|
|
search_results = wikipedia.search(search_term) |
|
return search_results |
|
|
|
|
|
|
|
def get_wiki_text(page_title): |
|
"""Grab text from a wikipedia page""" |
|
|
|
page = wikipedia.page(page_title) |
|
return page.content |
|
|
