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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,8 +9,8 @@ from transformers.pipelines import pipeline
9
 
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,
 
9
 
10
  classifier = pipeline("zero-shot-classification", model="knowledgator/comprehend_it-base")
11
 
12
+ candidate_labels = ['Lecture quality', 'lecturer teaching strength', 'assignments given', 'comments on lecturers teaching venue','Lecturers test and exam grading']
13
+ sequence_to_classify = "I hate Mr Thomas class, he doesn't teach well and he doesn't give assignments"
14
  st.write(classifier(sequence_to_classify, candidate_labels, multi_label=True))
15
  #{'labels': ['travel', 'exploration', 'dancing', 'cooking'],
16
  # 'scores': [0.9945111274719238,