hussain-shk commited on
Commit
338abb1
1 Parent(s): 48b0ee3

added label

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -6,12 +6,15 @@ roberta_pipe = pipeline(
6
  tokenizer="siebert/sentiment-roberta-large-english",
7
  return_all_scores = True
8
  )
 
9
  def analyse_sentiment(text):
10
  response = roberta_pipe(text)
11
- return response
 
 
 
12
 
13
- text = gr.inputs.Textbox(lines=5, placeholder="Enter Text to Get Sentiment",default="", label="Enter Text")
14
- text_output = gr.outputs.Textbox(type="auto", label=f"Sentiment")
15
 
16
- iface = gr.Interface(fn=analyse_sentiment, inputs=[text], outputs=text_output, title='Roberta Sentiment Analysis')
17
  iface.launch(enable_queue=True)
 
6
  tokenizer="siebert/sentiment-roberta-large-english",
7
  return_all_scores = True
8
  )
9
+
10
  def analyse_sentiment(text):
11
  response = roberta_pipe(text)
12
+ d = {}
13
+ for i in response[0]:
14
+ d[i['label'].lower()] = i['score']
15
+ return d
16
 
17
+ text = gr.Textbox(lines=5, placeholder="Enter Text to Get Sentiment",default="", label="Enter Text")
 
18
 
19
+ iface = gr.Interface(fn=analyse_sentiment, inputs=[text], outputs="label", title='Roberta Sentiment Analysis')
20
  iface.launch(enable_queue=True)