Update app.py

app.py

@@ -1,10 +1,10 @@
 import streamlit as st
-from transformers import pipeline
 
-pipe = pipeline('sentiment-analysis')
+with open('./trained_model.pkl', 'rb') as file:
+    loaded_model = pickle.load(file)
 
-text = st.text_area('Enter text')
+input = st.text_area('Enter text')
 
 if text:
-    out = pipe(text)
-    st.json(out)
+    out = loaded_model(input)
+    st.json(out[0][0])