Spaces:

OmParkashPandeY
/

TextSummarization

Build error

TextSummarization / src /controllers /text_summarize_controller.py

Upload 3 files

77fb0a9 almost 3 years ago

439 Bytes

import gradio as GUI
from transformers import pipeline
from ..services.text_summarize_service import completion_message

completion_message = pipeline("summarization", model='facebook/bart-large-cnn')


def summarize(input):
    output = completion_message(input)
    return output[0]['summary_text']
    
def loadGUI():
    GUI.close_all()
    demo = GUI.Interface(fn=summarize, inputs="text", outputs="text")
    demo.launch(share=True)


	import gradio as GUI
	from transformers import pipeline
	from ..services.text_summarize_service import completion_message

	completion_message = pipeline("summarization", model='facebook/bart-large-cnn')


	def summarize(input):
	output = completion_message(input)
	return output[0]['summary_text']

	def loadGUI():
	GUI.close_all()
	demo = GUI.Interface(fn=summarize, inputs="text", outputs="text")
	demo.launch(share=True)