ganeshkamath89's picture
passing share=True to make the application sharable publicly
85fa6c7 verified
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)