anegi commited on
Commit
f9a6a8c
1 Parent(s): 86e8910

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
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