chrismay commited on
Commit
6aacc74
·
1 Parent(s): 8a8dc07

Update app.py

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