Wootang01 commited on
Commit
e3a7576
·
1 Parent(s): 2199b57

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  st.title("Text Classifier - BART-Large-MNLI")
5
  st.write("Input text and labels. Submit. The machine will classify the text according to the labels.")
6
 
7
- classifier = pipeline(task="zero-shot-classification", device=0, model="facebook/bart-large-mnli")
8
 
9
  input_text = st.text_input(label='Input Text')
10
  labels = st.text_input(label='Labels')
 
4
  st.title("Text Classifier - BART-Large-MNLI")
5
  st.write("Input text and labels. Submit. The machine will classify the text according to the labels.")
6
 
7
+ classifier = pipeline(task="zero-shot-classification")
8
 
9
  input_text = st.text_input(label='Input Text')
10
  labels = st.text_input(label='Labels')