dancingninjas commited on
Commit
f90c3a5
·
1 Parent(s): 387bdf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -7
app.py CHANGED
@@ -3,16 +3,26 @@ from transformers import pipeline
3
 
4
  # Title and Subtitle
5
  st.title('Sentiment Analysis App')
 
6
 
7
- pipe=pipeline(model="dancingninjas/sentiment-model")
8
- text=st.text_area('Enter your text')
9
 
10
  if text:
11
- out = pipe(text)
 
 
12
 
13
- # Convert the model's output (0 or 1) to descriptive labels
14
- sentiment_label = "Negative" if out[0]['label'] == 'LABEL_1' else "Positive"
15
 
16
- # Display the sentiment label and probability
 
17
  st.write(f'Sentiment: {sentiment_label}')
18
- st.write(f'Probability: {out[0]["score"]:.4f}')
 
 
 
 
 
 
 
3
 
4
  # Title and Subtitle
5
  st.title('Sentiment Analysis App')
6
+ st.subheader('Enter text to analyze sentiment')
7
 
8
+ # Text input box
9
+ text = st.text_area('Enter your text')
10
 
11
  if text:
12
+ # Analyze sentiment using the model
13
+ pipe = pipeline(model="dancingninjas/sentiment-analysis-nlp")
14
+ results = pipe(text)
15
 
16
+ # Determine sentiment label
17
+ sentiment_label = "Positive" if results[0]['label'] == 'LABEL_1' else "Negative"
18
 
19
+ # Display sentiment label and probability
20
+ st.subheader('Sentiment Analysis Result:')
21
  st.write(f'Sentiment: {sentiment_label}')
22
+ st.write(f'Probability: {results[0]["score"]:.4f}')
23
+
24
+ # Visualize the sentiment with an emoji
25
+ if sentiment_label == 'Positive':
26
+ st.write('😃')
27
+ else:
28
+ st.write('😔')