saidsef commited on
Commit
11225f5
·
verified ·
1 Parent(s): df4609d

fix: add model and text area title

Browse files

- fix: add model and text area title

Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- pipe = pipeline("sentiment-analysis")
5
- text = st.text_area("what is the format of a report")
6
 
7
  if text:
8
  out = pipe(text)
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ pipe = pipeline(task="sentiment-analysis", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
5
+ text = st.text_area("what is the sentiment score?")
6
 
7
  if text:
8
  out = pipe(text)