import streamlit as st
import pickle
import sklearn
import pandas as pd
from sklearn.pipeline import Pipeline
st.title("Tags Predictor!")
st.write("Give your context...I will give you its related tags.")
text = st.text_input("Enter your Content:")
pre_proc = pickle.load(open("pipeline.pickle",'rb'))
mlb = pickle.load(open("mlb.pickle",'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))