File size: 284 Bytes
7f8de92 68826e9 a363694 e856b8c 71af36e |
1 2 3 4 5 6 7 8 9 10 |
import streamlit as st
from transformers import pipeline
st.title("Milestone #2")
model_name = st.text_input("which model do you want to use?")
classifier = pipeline("zero-shot-classification")
res = classifier("possible lables", cata = [offensive],)
print(res)
st.button("submit")
|