Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,6 +10,7 @@ from transformers.pipelines import pipeline
|
|
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))
|
14 |
#{'labels': ['travel', 'exploration', 'dancing', 'cooking'],
|
15 |
# 'scores': [0.9945111274719238,
|
|
|
10 |
classifier = pipeline("zero-shot-classification", model="knowledgator/comprehend_it-base")
|
11 |
|
12 |
candidate_labels = ['travel', 'cooking', 'dancing', 'exploration']
|
13 |
+
sequence_to_classify = "one day I will see the world"
|
14 |
st.write(classifier(sequence_to_classify, candidate_labels, multi_label=True))
|
15 |
#{'labels': ['travel', 'exploration', 'dancing', 'cooking'],
|
16 |
# 'scores': [0.9945111274719238,
|