Donlapark commited on
Commit
a718b41
·
1 Parent(s): 7855c89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -4
app.py CHANGED
@@ -1,7 +1,17 @@
1
  import streamlit as st
2
  from transformers import pipeline
 
3
  classifier = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
4
- text = st.text_area('enter some text:')
5
- if text:
6
- results = classifier([text])
7
- st.json(results)
 
 
 
 
 
 
 
 
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
+
4
  classifier = pipeline("sentiment-analysis", model="distilbert-base-uncased-finetuned-sst-2-english")
5
+ def main():
6
+ st.title("Adult Income")
7
+
8
+ with st.form("text_field"):
9
+ text = st.text_area('enter some text:')
10
+ # clicked==True only when the button is clicked
11
+ clicked = st.form_submit_button("Predict income")
12
+ if clicked:
13
+ results = classifier([text])
14
+ st.json(results)
15
+
16
+ if __name__ == "__main__":
17
+ main()