Update app.py

app.py

@@ -1,18 +1,16 @@
 import streamlit as st
 from transformers import pipeline
-from PIL import Image
 
-classifier = pipeline("image-classification", model="google/vit-base-patch16-224-in21k")
+classifier = pipeline("text-classification", model="distilbert-base-uncased-finetuned-sst-2-english")
 def main():
     st.title("text-classification")
 
     with st.form("text_field"):
-        text = st.file_uploader(label = "Upload your image here",type=['png','jpg','jpeg'])
+        text = st.text_area('enter some text:')
         # clicked==True only when the button is clicked
         clicked = st.form_submit_button("Submit")
         if clicked:
-          img=Image.open(text)
-          results = classifier([img])
+          results = classifier([text])
           st.json(results)
 
 if __name__ == "__main__":