speechemotion / app.py
Nattyboi's picture
Update app.py
1d52a8c
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))