| """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 |
|
|
