ganeshkamath89's picture
adding model attribute and specifying a model that is capable of taking larger texts
94e1082 verified
raw
history blame
975 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('MyProjectName ([email protected])', '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]
summarizer = pipeline("summarization", model = "allenai/led-large-16384-arxiv")
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()