File size: 289 Bytes
7f8de92
 
68826e9
1188df8
854d0de
 
 
 
 
2cd8647
71af36e
1
2
3
4
5
6
7
8
9
10
11
12
import streamlit as st
from transformers import pipeline 
st.title("Milestone #2")
text = st.text_input("write a statement")
con = st.button("submit")
if con:
  classifier = pipeline("zero-shot-classification")
  res = classifier(text, cata = ["offensive"],)
  print(res)
  st.write(res)