Update app.py

app.py

@@ -1,9 +1,11 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import Automodel
+import pickle
+from tensorflow.keras.models import load_model
 
-pipe = pipeline('sentiment-analysis')
+model = load_model('tox_model.pkl')
 text = st.text_area('Enter some text')
 
 if text:
-  out = pipe(text)
+  out = model.predict(text)
   st.json(out)