Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,14 @@
 import streamlit as st
+import pickle
+import sklearn
+import pandas as pd
+from sklearn.transformers import Pipeline
 st.title("Tags Predictor!")
 st.write("Give your context...I will give you its related tags.")
-st.text_input("Enter your Content:")
-#pre_proc = pickle.load()
+text = st.text_input("Enter your Content:")
+pre_proc = pickle.load(open("pipeline.pkl",'rb'))
+mlb = pickle.load(open("mlb.pkl",'rb'))
+cleaned = pre_proc.named_steps['pre_processing'].transform(pd.Series([text]))
+vector = pre_proc.named_steps['vectorizer'].transform(cleaned)
+prediction = pre_proc.named_steps['classifier'].predict(vector)
+print(mlb.inverse_transform(prediction))