Spaces:
Runtime error
Runtime error
#!/usr/bin/env python | |
# coding: utf-8 | |
# In[1]: | |
import gradio as gr | |
from transformers import pipeline | |
import requests | |
from bs4 import BeautifulSoup | |
# In[9]: | |
pipe = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6") | |
def get_abstract(url): | |
page = requests.get(url) | |
soup = BeautifulSoup(page.content, "html.parser") | |
abstract = soup.find(id="eng-abstract").text | |
return abstract | |
def summarize(input): | |
abstract = get_abstract(input) | |
summary = pipe(abstract) | |
return summary[0]["summary_text"] | |
demo = gr.Interface( | |
fn=summarize, | |
inputs=gr.Textbox(placeholder="PubMed URL", label="PubMed URL"), | |
outputs=gr.Textbox(placeholder="Your Summary will appear here", label="Summary"), | |
title="PubMed Summarizer📝", | |
examples=["https://pubmed.ncbi.nlm.nih.gov/36258852/", "https://pubmed.ncbi.nlm.nih.gov/34747661/"]) | |
demo.launch() | |
# In[ ]: | |