Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,8 +6,8 @@ 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.pkl",'rb'))
-mlb = pickle.load(open("mlb.pkl",'rb'))
+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)