File size: 571 Bytes
1a80035
 
8f8b328
 
 
 
 
 
79b6ab8
 
 
 
1a80035
79b6ab8
8f8b328
79b6ab8
 
8f8b328
79b6ab8
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import streamlit as st

from transformers import pipeline


classifier = pipeline("zero-shot-classification",
                      model="valhalla/distilbart-mnli-12-1")

with st.form('inputs') as form:
    input_text = form.text_area("Enter your sample")
    input_label = form.text_input("Enter labels", placeholder="support, help, important")
    submit_button = form.form_submit_button(label='Submit')

if submit_button:

    labels = list(l.strip() for l in input_label.split(','))
    pred = classifier(input_text, labels, multi_class=True)

    st.markdown(pred)