chrismay commited on
Commit
8a8dc07
·
1 Parent(s): 4b57d87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -1,9 +1,16 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- pipe = pipeline('sentiment-analysis')
5
  text = st.text_area('Enter text here!')
6
 
 
 
 
 
 
 
 
7
  if text:
8
  out = pipe(text)
9
  st.json(out)
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ # pipe = pipeline('sentiment-analysis')
5
  text = st.text_area('Enter text here!')
6
 
7
+ classifier = pipeline("zero-shot-classification",
8
+ model="facebook/bart-large-mnli")
9
+
10
+ sequence_to_classify = text
11
+ candidate_labels = ['Positive', 'Neutral', 'Negative']
12
+ pipe = classifier(sequence_to_classify, candidate_labels)
13
+
14
  if text:
15
  out = pipe(text)
16
  st.json(out)