Update app.py

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
 from transformers import pipeline 
 st.title("Milestone #2")
-model_name = st.text_input("which model do you want to use?")
+text = st.text_input("write a statement")
 classifier = pipeline("zero-shot-classification")
-res = classifier("possible lables", cata = ["offensive"],)
+res = classifier(text, cata = ["offensive"],)
 print(res)
 st.button("submit")