import streamlit as st


from transformers.pipelines import pipeline


# Use a pipeline as a high-level helper


classifier = pipeline("zero-shot-classification", model="knowledgator/comprehend_it-base")

candidate_labels = ['Lecture quality','Positive lecture quality','Negative lecture quality', 'lecturer teaching strength', 'assignments not given', 'assignments given',' positve comments on lecturers teaching venue','neutral comments on lecturers teaching venue','negative comments on lecturers teaching venue',' Lecturers test and exam grading']
sequence_to_classify = "I hate Mr Thomas class, he doesn't teach well and he doesn't give assignments"
st.write(classifier(sequence_to_classify, candidate_labels, multi_label=True))