abdoulayee commited on
Commit
a95ef48
·
verified ·
1 Parent(s): a3e1d08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -0
app.py CHANGED
@@ -1,3 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  # Déployer
2
  import gradio as gr
3
  import os
 
1
+ # Importer pipeline
2
+ from transformers import pipeline
3
+ # charger le modèle bart-large-cnn
4
+ summarizer = pipeline(task="summarization", model="facebook/bart-large-cnn")
5
+ # charger le modèle mT5_multilingual_XLSum
6
+ summarizer_1= pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum")
7
+ # Définir une fonction summarize_func avec bart_large_cnn
8
+ def summarize_func(input, min_length, max_length):
9
+ output = summarizer(input.strip(),min_length, max_length)
10
+ return output[0]['summary_text']
11
+
12
+ # Définir une fonction summarize_func ave cmT5-multilingual
13
+ def summarize_func_1(input, min_length, max_length):
14
+ output = summarizer_1(input.strip(), min_length, max_length)
15
+ return output[0]['summary_text']
16
  # Déployer
17
  import gradio as gr
18
  import os