Spaces:
Sleeping
Sleeping
File size: 590 Bytes
ee2e7b3 482523d 3c4cca2 c35d804 482523d db38ad9 482523d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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))
|