|
import requests, json |
|
|
|
|
|
def get_completion(inputs, parameters=None,ENDPOINT_URL='https://api-inference.huggingface.co/models/DunnBC22/flan-t5-base-text_summarization_data'): |
|
|
|
|
|
headers = { |
|
"Content-Type": "application/json" |
|
} |
|
data = { "inputs": inputs } |
|
if parameters is not None: |
|
data.update({"parameters": parameters}) |
|
response = requests.request("POST", |
|
ENDPOINT_URL, headers=headers, |
|
data=json.dumps(data) |
|
) |
|
return json.loads(response.content.decode("utf-8")) |
|
|
|
|
|
import gradio as gr |
|
|
|
def summarize(input): |
|
output = get_completion(input) |
|
return output[0]['summary_text'] |
|
|
|
gr.close_all() |
|
demo = gr.Interface(fn=summarize, |
|
inputs=[gr.Textbox(label="Text to summarize", lines=6)], |
|
outputs=[gr.Textbox(label="Result", lines=3)], |
|
title="Text summarization with distilbart-cnn", |
|
description="Summarize any text using the `shleifer/distilbart-cnn-12-6` model under the hood!" |
|
) |
|
demo.launch(inline=False) |
|
|
|
|