andrea777h commited on
Commit
b27870c
1 Parent(s): b2ca231

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -0
app.py ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+ import gradio as gr
3
+
4
+ sumirizer=pipeline("summarization",model="t5-base",tokenizer="t5-small",truncation=True,framework="tf")
5
+
6
+ def translate(text):
7
+ text=text.replace('"','"')
8
+ text=text.replace(''',"'")
9
+ text=text.replace('&','&')
10
+ result=summarizer(text, min_length=180,truncation=True,framework="tf")
11
+ return result[0]["summary_text"]
12
+
13
+ iface=gr.Interface(
14
+ fn=translate,
15
+ inputs=gr.inputs.Textbox(lines=10, placeholder="Enter text to summarize..."),
16
+ outputs='text'
17
+ )
18
+ iface.lunch()