Create summarize.py

summarize.py

@@ -0,0 +1,29 @@
+"""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"]
+
+
+# search wikipedia pages
+def search_wiki(search_term):
+    """Search wikipedia pages"""
+
+    search_results = wikipedia.search(search_term)
+    return search_results
+
+
+# grab text from a wikipedia page
+def get_wiki_text(page_title):
+    """Grab text from a wikipedia page"""
+
+    page = wikipedia.page(page_title)
+    return page.content