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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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,