ganeshkamath89's picture
Trying out a different code to get Wikipedia Summary
cb38464 verified
raw
history blame
784 Bytes
from transformers import pipeline
import gradio as gr
import wikipediaapi
import nltk
nltk.download('punkt')
def get_wiki_summary(search):
wiki_wiki = wikipediaapi.Wikipedia('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
summary = page.summary[0:60]
text = page.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"),
gr.Text(label="Text")
]
)
wiki_summary.launch()