Margaret commited on
Commit
68a6600
1 Parent(s): 7a61b46

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -8
app.py CHANGED
@@ -1,16 +1,13 @@
1
  import gradio as gr
2
-
3
  from transformers import pipeline
4
 
5
- pipe = pipeline("sentiment-analysis", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
6
-
7
  def get_sentiment(input_text):
8
- return pipe(input_text)[0]["label"]
9
 
10
  iface = gr.Interface(fn = get_sentiment,
11
  inputs = "text",
12
- outputs = 'text',
13
  title= 'Sentiment Analysis',
14
- description = 'Get Sentiment Negative/Positive/Neutral for the given input')
15
-
16
- iface.launch(inline = False)
 
1
  import gradio as gr
 
2
  from transformers import pipeline
3
 
4
+ sentiment = pipeline("sentiment-analysis", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
 
5
  def get_sentiment(input_text):
6
+ return sentiment(input_text)
7
 
8
  iface = gr.Interface(fn = get_sentiment,
9
  inputs = "text",
10
+ outputs = ['text'],
11
  title= 'Sentiment Analysis',
12
+ description = 'Get Sentiment Negative/Positive for the given input')
13
+ iface.launch(inline = False)