Gebremichael commited on
Commit
8dc027e
·
1 Parent(s): d25d22e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -5
app.py CHANGED
@@ -1,9 +1,20 @@
1
- import streamlit as str
2
  from transformers import pipeline
 
3
 
4
  pipe = pipeline('sentiment-analysis')
5
- text = st.text_area('enter some text!')
 
 
 
 
 
 
6
 
7
- if text:
8
- out = pipe(text)
9
- st.json(out)
 
 
 
 
 
1
+ import streamlit as st
2
  from transformers import pipeline
3
+ from textblob import TextBlob
4
 
5
  pipe = pipeline('sentiment-analysis')
6
+ st.title("Hugging Face Sentiment Analysis Spaces Example")
7
+ st.subheader("What framework would you like to use for Sentiment Analysis")
8
+ #Picking what NLP task you want to do
9
+ option = st.selectbox('Framework',('Transformers', 'TextBlob')) #option is stored in this variable
10
+ #Textbox for text user is entering
11
+ st.subheader("Enter the text you'd like to analyze.")
12
+ text = st.text_input('Enter text') #text is stored in this variable
13
 
14
+ if option == 'Transformers':
15
+ out = pipe(text)
16
+ else:
17
+ out = TextBlob(text)
18
+ out = out.sentiment
19
+ st.write("Sentiment of Text: ")
20
+ st.write(out)