hussain-shk commited on
Commit
8e145a2
·
1 Parent(s): 7be5f85

added sentiment

Browse files
Files changed (1) hide show
  1. app.py +39 -10
app.py CHANGED
@@ -15,14 +15,43 @@ INDIC = {"Assamese": "as", "Bengali": "bn", "Gujarati": "gu", "Hindi": "hi","Kan
15
  def translate(text, lang):
16
  return indic2en_model.translate_paragraph(text, INDIC[lang], 'en')
17
 
18
- languages = list(INDIC.keys())
19
- drop_down = gr.inputs.Dropdown(languages, type="value", default="Hindi", label="Select Source Language")
20
- text = gr.inputs.Textbox(lines=5, placeholder="Enter Text to translate", default="", label="Enter Text in Source Language")
21
- text_ouptut = gr.outputs.Textbox(label="Translated text in English")
22
 
23
- # example=[['I want to translate this sentence in Hindi','Hindi'],
24
- # ['I am feeling very good today.', 'Bengali']]
25
-
26
- supported_lang = ', '.join(languages)
27
- iface = gr.Interface(enable_queue=True,fn=translate, inputs=[text,drop_down] , outputs=text_ouptut, title='IndicTrans NMT System', description = 'Currently the model supports ' + supported_lang , examples=None)
28
- iface.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
  def translate(text, lang):
16
  return indic2en_model.translate_paragraph(text, INDIC[lang], 'en')
17
 
 
 
 
 
18
 
19
+ from transformers import pipeline
20
+ import gradio as gr
21
+ roberta_pipe = pipeline(
22
+ "sentiment-analysis",
23
+ model="siebert/sentiment-roberta-large-english",
24
+ tokenizer="siebert/sentiment-roberta-large-english",
25
+ return_all_scores = True
26
+ )
27
+
28
+ def analyse_sentiment(text, source):
29
+ if source != "English":
30
+ text = translate(text, source)
31
+ response = roberta_pipe(text)
32
+ d = {}
33
+ for i in response[0]:
34
+ d[i['label'].lower()] = i['score']
35
+ return d
36
+
37
+ languages = ["Assamese", "Bengali", "Gujarati", "Hindi", "Kannada","Malayalam", "Marathi", "Odia", "Punjabi", "Tamil", "Telugu", "English"]
38
+
39
+ input_text = gr.Textbox(placeholder="Enter a positive or negative sentence here...")
40
+ drop_down = gr.inputs.Dropdown(languages, type="value", default="English", label="Select Source Language")
41
+
42
+ examples = [["this book was a great book that i have read many times", "English"],
43
+ ["एक महान अमेरिकी लेखक का एक आकर्षक संग्रह" , "Hindi"],
44
+ ["हा आतापर्यंतचा सर्वात वाईट चित्रपट आहे यात शंका नाही", "Marathi"],
45
+ ["இந்த தயாரிப்பு ஆச்சரியமாக இருக்கிறது", "Tamil"],
46
+ ["તમારા માટે નહીં જો તમે વિના અવરોધે વીડિયો શોધી રહ્યા છો", "Gujarati"],]
47
+
48
+ demo = gr.Interface(
49
+ enable_queue=True,
50
+ fn=analyse_sentiment,
51
+ inputs=[input_text, drop_down],
52
+ outputs="label",
53
+ interpretation="default",
54
+ title='IndiSent: Multilingual Sentiment Analysis',
55
+ examples=examples)
56
+
57
+ demo.launch(share = True, debug=True, inline=True)