Spaces:

TanmayP
/

TextSummarizer

Runtime error

Update app.py

a2ffb96 verified almost 2 years ago

1.97 kB

import torch
import gradio
from transformers import pipeline

# Used the model ID from the Hugging Face Hub
text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",torch_dtype=torch.bfloat16)

#Example text
# text = '''Costello's (also known as Tim's) was a bar and restaurant in Midtown Manhattan,
#     New York City, from 1929 to 1992. The bar operated at several locations near the intersection
#  of East 44th Street and Third Avenue. Costello's was known as a drinking spot for journalists with the
#   New York Daily News, writers with The New Yorker, novelists, and cartoonists, including the author Ernest Hemingway,
#   the cartoonist James Thurber, the journalist John McNulty, the poet Brendan Behan, the short-story writer John O'Hara,
#   and the writers Maeve Brennan and A. J. Liebling. The bar is also known for having been home to a wall where Thurber drew a
#   cartoon depiction of the "Battle of the Sexes" at some point between 1934 and 1935; the cartoon was destroyed, illustrated again,
#   and then lost in the 1990s. A wall illustrated in 1976 by several cartoonists, including Bill Gallo, Stan Lee, Mort Walker,
#    Al Jaffee, Sergio Aragonés, and Dik Browne, is still on display at the bar's final location.'''
#
# # Print the summary
# print(text_summary(text))


#we are writing this function gardio accepts function, this function will return the output
def summary(input):
    output = text_summary(input)
    return output[0]['summary_text']


gradio.close_all()

# demo = gradio.Interface(fn=summuray,inputs="text",outputs="text")

demo = gradio.Interface(fn=summary,
                        inputs=[gradio.Textbox(label="Input text to summarize",lines=6)],
                        outputs=[gradio.Textbox(label="Summarized text", lines=4)],
                        title="@GenAi Project 1: Text Summarizer",
                        description="This application can used for summarizing the text")


demo.launch()

	import torch
	import gradio
	from transformers import pipeline

	# Used the model ID from the Hugging Face Hub
	text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",torch_dtype=torch.bfloat16)

	#Example text
	# text = '''Costello's (also known as Tim's) was a bar and restaurant in Midtown Manhattan,
	# New York City, from 1929 to 1992. The bar operated at several locations near the intersection
	# of East 44th Street and Third Avenue. Costello's was known as a drinking spot for journalists with the
	# New York Daily News, writers with The New Yorker, novelists, and cartoonists, including the author Ernest Hemingway,
	# the cartoonist James Thurber, the journalist John McNulty, the poet Brendan Behan, the short-story writer John O'Hara,
	# and the writers Maeve Brennan and A. J. Liebling. The bar is also known for having been home to a wall where Thurber drew a
	# cartoon depiction of the "Battle of the Sexes" at some point between 1934 and 1935; the cartoon was destroyed, illustrated again,
	# and then lost in the 1990s. A wall illustrated in 1976 by several cartoonists, including Bill Gallo, Stan Lee, Mort Walker,
	# Al Jaffee, Sergio Aragonés, and Dik Browne, is still on display at the bar's final location.'''
	#
	# # Print the summary
	# print(text_summary(text))


	#we are writing this function gardio accepts function, this function will return the output
	def summary(input):
	output = text_summary(input)
	return output[0]['summary_text']


	gradio.close_all()

	# demo = gradio.Interface(fn=summuray,inputs="text",outputs="text")

	demo = gradio.Interface(fn=summary,
	inputs=[gradio.Textbox(label="Input text to summarize",lines=6)],
	outputs=[gradio.Textbox(label="Summarized text", lines=4)],
	title="@GenAi Project 1: Text Summarizer",
	description="This application can used for summarizing the text")


	demo.launch()