TextSummary / app.py
Simran-Dalvi's picture
Update app.py
a80f69f verified
import torch
import gradio as gr
# Use a pipeline as a high-level helper
from transformers import pipeline
text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
# model_path ="../Models/models--sshleifer--distilbart-cnn-12-6/snapshots/a4f8f3ea906ed274767e9906dbaede7531d660ff"
# text_summary = pipeline("summarization", model=model_path,
# torch_dtype=torch.bfloat16)
# text=''Elon Reeve Musk (/ˈiːlɒn mʌsk/; born June 28, 1971) is a businessman and United States
# special government employee known for his key roles in the automotive company Tesla, Inc.
# and the space company SpaceX. He is also known for his ownership of the technology company
# X Corp. and his role in the founding of the Boring Company, xAI, Neuralink, and OpenAI.
# Musk is the wealthiest individual in the world; as of January 2025, Forbes estimates his net
# worth to be US$426 billion. Musk has engaged in political activities in several countries, including
# as a vocal and financial supporter of U.S. president Donald Trump. He became the largest
# donor in the 2024 United States presidential election, and a supporter of international
# far-right political parties, activists, and causes.
#
# A member of the wealthy South African Musk family, Musk was born in Pretoria before immigrating to
# Canada, acquiring its citizenship through his Canadian-born mother, Maye. Two years later, he matriculated
# at Queen's University in Canada. Musk later transferred to the University of Pennsylvania and received bachelor's
# degrees in economics and physics. He moved to California in 1995 to attend Stanford University, and with his brother
# Kimbal co-founded the software company Zip2, that was later acquired by Compaq in 1999. That same year, Musk co-founded
# X.com, a direct bank, that later formed PayPal. In 2002, Musk acquired United States citizenship, and eBay acquired PayPal.
# Using the money he made from the sale, Musk founded SpaceX, a spaceflight services company, in 2002. ''
# print(text_summary(text))
def summary(inputtext):
output= text_summary(inputtext)
return output[0]['summary_text']
gr.close_all()
demo = gr.Interface(fn=summary, inputs="text", outputs="text")
# demo = gr.Interface(fn=summary,
# inputs=[gr.Textbox(label="Input text to summarize",lines=6)],
# outputs=[gr.Textbox(label="Summarized text",lines=4)],
# title="GenAI Project 1: Text Summarizer",
# description="THIS APPLICATION IS USED TO SUMMARIZE TEXT")
demo.launch()