Hanorii commited on
Commit
b30f013
·
1 Parent(s): 52a9632

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -1,7 +1,10 @@
1
  import streamlit as st
2
 
3
- from transformers import AutoModel, AutoTokenizer
4
 
5
- # Load the ALBERT-base-v1 model and tokenizer
6
- model = AutoModel.from_pretrained("albert-base-v1")
7
- tokenizer = AutoTokenizer.from_pretrained("albert-base-v1")
 
 
 
 
1
  import streamlit as st
2
 
3
+ from transformers import pipeline
4
 
5
+ pipe = pipeline("sentiment-analysis")
6
+ text = st.text_area("enter some text")
7
+ if text:
8
+ out= pipe(text)
9
+ st.json(out)
10
+