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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from transformers.pipelines import pipeline
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'],
 
9
 
10
  classifier = pipeline("zero-shot-classification", model="knowledgator/comprehend_it-base")
11
 
12
+ candidate_labels = ['Lecture quality','Positive lecture quality','Negative 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'],