krrishD/google_pegasus-large

1import gradio as gr2from transformers import pipeline3 4pipe = pipeline("summarization", model="google/pegasus-large")5 6def main(in_text):7  print(in_text)8  answer = pipe(in_text, min_length=5, max_length=20)9  print(answer)10  return answer[0]["summary_text"]11 12with gr.Blocks() as demo:13  gr.Markdown("""# Summarization Engine!""")14  with gr.Row():15    with gr.Column():16      text1 = gr.Textbox(17            label="Input Text",18            lines=1,19        )20      output = gr.Textbox(label="Output Text")21      b1 = gr.Button("Summarize!")22      b1.click(main, inputs=[text1], outputs=output)23  gr.Markdown("""#### powered by [Tassle](https://bit.ly/3LXMklV)""")24 25 26if __name__ == "__main__":27    demo.launch(debug=True)