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)