iamomtiwari commited on
Commit
35a8dfb
·
verified ·
1 Parent(s): 4d84494

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -0
app.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from pipeline import GOTSummarizer
3
+
4
+ summarizer = GOTSummarizer()
5
+
6
+ def process(text, lang='en'):
7
+ summary = summarizer.summarize(text)
8
+ if lang != 'en':
9
+ return summarizer.translate(summary, lang)
10
+ return summary
11
+
12
+ gr.Interface(
13
+ fn=process,
14
+ inputs=[gr.Textbox(lines=10), gr.Dropdown(['en', 'hi', 'pa', 'ta', 'bn'])],
15
+ outputs="text"
16
+ ).launch()