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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -1,13 +1,16 @@
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)
 
 
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()