speechemotion / app.py
Nattyboi's picture
Update app.py
0fd3349
raw
history blame
691 Bytes
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))