Spaces:
Runtime error
Runtime error
| import json | |
| import gradio as gr | |
| from transformers import AutoModelWithLMHead, AutoTokenizer | |
| from transformers import AutoTokenizer, AutoModelForSeq2SeqLM | |
| from transformers import pipeline | |
| def summarize_conversation(conversation, slider_value): | |
| summarizer = pipeline("summarization", model="philschmid/bart-large-cnn-samsum") | |
| summarized_conversation = summarizer(conversation.strip(), min_length=int(slider_value)) | |
| return summarized_conversation[0]["summary_text"] | |
| def change_textbox(choice): | |
| with open('examples.json') as json_file: | |
| examples = json.load(json_file) | |
| if choice == "Customer Dialog AmazonHelp": | |
| return gr.update(lines=8, visible=True, value=str(examples["dialog_1"])) | |
| elif choice == "Customer Dialog XBoxSupport": | |
| return gr.update(lines=8, visible=True, value=str(examples["dialog_2"])) | |
| elif choice == "Customer Dialog AppleSupport": | |
| return gr.update(lines=8, visible=True, value=str(examples["dialog_3"])) | |
| elif choice == "Fridge: Ice maker is not working": | |
| return gr.update(lines=8, visible=True, value=str(examples["text_ice_maker_not_working"])) | |
| elif choice == "Troubleshooting Iphone": | |
| return gr.update(lines=8, visible=True, value=str(examples["text_iphone_troubleshooting"])) | |
| else: | |
| return gr.update(lines=8, visible=True, value="") | |
| with gr.Blocks() as demo: | |
| radio = gr.Radio( | |
| ["Free Text Input", "Customer Dialog AmazonHelp", "Customer Dialog XBoxSupport", "Customer Dialog AppleSupport", "Fridge: Ice maker is not working", "Troubleshooting Iphone"], label="What should the AI summarize?" | |
| ) | |
| text = gr.Textbox(lines=2, interactive=True, placeholder="Your text here...") | |
| output = gr.Textbox(lines=2, interactive=True, placeholder="Your result here...") | |
| radio.change(fn=change_textbox, inputs=radio, outputs=text) | |
| slider_value = gr.Slider(minimum=20, maximum=142, value=42, randomize=False, label="Minimal length of summary.") | |
| button_clear = gr.Button("Clear") | |
| button_1 = gr.Button("Summarize") | |
| button_1.click(fn=summarize_conversation, inputs=[text, slider_value], outputs=output) | |
| button_clear.click( | |
| lambda: [c.update(value=None) for c in [text, output]], | |
| inputs=[], | |
| outputs=[text, output] | |
| ) | |
| demo.launch() |