Nattyboi commited on
Commit
b560784
·
1 Parent(s): bf71f96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,4 +1,4 @@
1
- import streamlit
2
 
3
 
4
  from transformers.pipelines import pipeline
@@ -7,7 +7,7 @@ from transformers.pipelines import pipeline
7
  # Use a pipeline as a high-level helper
8
 
9
 
10
- pipe = pipeline("zero-shot-classification", model="knowledgator/comprehend_it-base")
11
 
12
  candidate_labels = ['travel', 'cooking', 'dancing', 'exploration']
13
  st.write(classifier(sequence_to_classify, candidate_labels, multi_label=True))
 
1
+ import streamlit as st
2
 
3
 
4
  from transformers.pipelines import pipeline
 
7
  # Use a pipeline as a high-level helper
8
 
9
 
10
+ classifier = pipeline("zero-shot-classification", model="knowledgator/comprehend_it-base")
11
 
12
  candidate_labels = ['travel', 'cooking', 'dancing', 'exploration']
13
  st.write(classifier(sequence_to_classify, candidate_labels, multi_label=True))