meshsl commited on
Commit
cf95e39
·
verified ·
1 Parent(s): 54e0967

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +55 -0
app.py ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import warnings
2
+ warnings.filterwarnings('ignore')
3
+ import transformers
4
+ import gradio as gr
5
+ from transformers import pipeline
6
+ from TTS.api import TTS
7
+
8
+ #1 - Sentiment Analysis Tab:
9
+ sentiment =pipeline("text-classification", model="tabularisai/multilingual-sentiment-analysis")
10
+
11
+ def get_sentiment(text):
12
+ result = sentiment(text)
13
+ label = result[0]['label']
14
+ score = result[0]['score']
15
+ return label, score
16
+
17
+ #2 - Summarization Tab:
18
+ summary_pipe = pipeline("summarization", model="facebook/bart-large-cnn")
19
+
20
+ def get_summary(text):
21
+ result = summary_pipe(text[:1000])
22
+ return result[0]['summary_text']
23
+
24
+ #3 - text-to-Speech Tab:
25
+ tts_model= TTS("tts_models/en/ljspeech/tacotron2-DDC", gpu=False)
26
+
27
+ def text_to_speech(text):
28
+ filename = "output.wav"
29
+ tts_model.tts_to_file(text=text, file_path=filename)
30
+ return filename
31
+
32
+
33
+ with gr.Blocks() as AI_applications:
34
+ with gr.Tab("Sentiment Analysis"):
35
+ sentiment_input = gr.Textbox(label="Enter Text")
36
+ output_label = gr.Text(label="Sentiment Label")
37
+ output_score = gr.Text(label="Confidence Score")
38
+ analyze_button = gr.Button("Analyze Sentiment")
39
+ analyze_button.click(fn=get_sentiment,inputs=sentiment_input ,outputs=[output_label, output_score])
40
+
41
+ with gr.Tab("Summarization"):
42
+
43
+ summary_input = gr.Textbox(label="Enter Long Text", lines=10, placeholder="Paste your text here...")
44
+ output_summary = gr.Textbox(label="Summary", lines=5)
45
+ summarize_btn = gr.Button("Summarize")
46
+ summarize_btn .click(fn=get_summary, inputs=summary_input, outputs=output_summary)
47
+
48
+ with gr.Tab("Text to Speech"):
49
+
50
+ tts_input = gr.Textbox(label="Enter Text to Speak", lines=4, placeholder="Type something...")
51
+ output_audio = gr.Audio(label="Generated Speech", type="filepath")
52
+ generate_btn = gr.Button("Generate Speech")
53
+ generate_btn.click(fn=text_to_speech, inputs=tts_input , outputs=output_audio)
54
+
55
+ AI_applications.launch()