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','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))