Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -4,6 +4,12 @@ from gradio.mix import Parallel
|
|
| 4 |
description = 'Dialogue summarization models'
|
| 5 |
title = 'Comparing different dialogue summarization models'
|
| 6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
model1 = gr.Interface.load("huggingface/philschmid/bart-large-cnn-samsum",
|
| 9 |
title = 'BART-Large-cnn-samsum'
|
|
@@ -17,7 +23,8 @@ model2 = gr.Interface.load("huggingface/anegi/t5smallmodel",
|
|
| 17 |
Parallel(model1, model2 ,
|
| 18 |
title = title,
|
| 19 |
inputs = gr.inputs.Textbox(lines = 7, label = 'Input Text', placeholder = 'Please enter your dialogue text here'),
|
| 20 |
-
|
|
|
|
| 21 |
theme = 'dark-peach' ).launch()
|
| 22 |
|
| 23 |
|
|
|
|
| 4 |
description = 'Dialogue summarization models'
|
| 5 |
title = 'Comparing different dialogue summarization models'
|
| 6 |
|
| 7 |
+
example = [['''A : Hello man. How are you?
|
| 8 |
+
B: I'm good. How's your course going these days?
|
| 9 |
+
A: Its going well. Pretty tiring byt overall fun.
|
| 10 |
+
B: That's great to hear. Wanna grab a cup of coffee?
|
| 11 |
+
A: Sure! I recently discovered a new cafe near my home.
|
| 12 |
+
B: Awesome. Let's go.''']]
|
| 13 |
|
| 14 |
model1 = gr.Interface.load("huggingface/philschmid/bart-large-cnn-samsum",
|
| 15 |
title = 'BART-Large-cnn-samsum'
|
|
|
|
| 23 |
Parallel(model1, model2 ,
|
| 24 |
title = title,
|
| 25 |
inputs = gr.inputs.Textbox(lines = 7, label = 'Input Text', placeholder = 'Please enter your dialogue text here'),
|
| 26 |
+
outputs= gr.outputs.Textbox(type="auto",label='Summary generated by the model'),
|
| 27 |
+
example = example,
|
| 28 |
theme = 'dark-peach' ).launch()
|
| 29 |
|
| 30 |
|