Spaces:
Sleeping
Sleeping
| from transformers import pipeline | |
| import gradio as gr | |
| import wikipediaapi | |
| import nltk | |
| nltk.download('punkt') | |
| def get_wiki_summary(search): | |
| wiki_wiki = wikipediaapi.Wikipedia('MyProjectName (merlin@example.com)', 'en') | |
| page = wiki_wiki.page(search) | |
| isExist = page.exists() | |
| if not isExist: | |
| return isExist, "Not found", "Not found", "Not found", "Not found" | |
| url = page.fullurl | |
| tittle = page.title | |
| text = page.text | |
| #summary = page.summary[0:300] | |
| summary = page.summary | |
| #summarizer = pipeline("summarization", model = "BeenaSamuel/t5_small_cnn_multi_news_abstractive_summarizer_v2") | |
| #summary = summarizer(text)[0]["summary_text"] | |
| return isExist, url, tittle, summary, text | |
| wiki_summary = gr.Interface( | |
| get_wiki_summary, | |
| gr.Text(label="Search Wikipedia"), | |
| [ | |
| gr.Text(label="Page exists?"), | |
| gr.Text(label="URL"), | |
| gr.Text(label="Title"), | |
| gr.Text(label="Summary", lines=4), | |
| gr.Text(label="Text", lines=10) | |
| ] | |
| ) | |
| wiki_summary.launch(share=True) |