Balde-maria2005 commited on
Commit
e535d30
·
verified ·
1 Parent(s): 3ef77c7

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +39 -0
app.py ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Définir une fonction summarize_func avec bart-large-cnn
2
+ def summarize_func(input, min_length, max_length):
3
+ output = summarizer(input.strip(),min_length, max_length)
4
+ return output[0]['summary_text']
5
+
6
+ # Définir une fonction summarize_func avec mT5-multilingual
7
+ def summarize_func_1(input, min_length, max_length):
8
+ output = summarizer_1(input.strip(), min_length, max_length)
9
+ return output[0]['summary_text']
10
+
11
+ # Déployer
12
+ import gradio as gr
13
+ # Création de blocks
14
+ demo = gr.Blocks(theme='NoCrypt/miku')
15
+ inputs = gr.Textbox(label="Text à résumer", lines=6)
16
+ summarizer1 = gr.Interface(fn=summarize_func,
17
+ inputs=inputs,
18
+ outputs=[gr.Textbox(label="Résumé", lines=3)],
19
+ title="Text summarization avec bart-large-cnn",
20
+ description="Résumer n'importe quel texte avec bart-large-cnn"
21
+ )
22
+ inputs1 = gr.Textbox(label="Text à résumer", lines=6)
23
+
24
+ summarizer2 = gr.Interface(fn=summarize_func_1,
25
+ inputs=inputs1,
26
+ outputs=[gr.Textbox(label="Result", lines=3)],
27
+ title="Text summarization avec mT5_multilingual_XLSum",
28
+ description="Résumer n'importe quel texte mT5_multilingual_XLSum"
29
+ )
30
+
31
+ with demo:
32
+ gr.TabbedInterface(
33
+ [summarizer1,
34
+ summarizer2],
35
+ ["Summarize avec mT5",
36
+ "Summarize avec bart"],
37
+ )
38
+
39
+ demo.launch()