Update Create_ml_models.py

Create_ml_models.py

@@ -6,6 +6,20 @@ from sklearn.preprocessing import StandardScaler
 import joblib
 from admin_utils import *
 
+if 'cleaned_data' not in st.session_state:
+    st.session_state['cleaned_data'] =''
+if 'sentences_train' not in st.session_state:
+    st.session_state['sentences_train'] =''
+if 'sentences_test' not in st.session_state:
+    st.session_state['sentences_test'] =''
+if 'labels_train' not in st.session_state:
+    st.session_state['labels_train'] =''
+if 'labels_test' not in st.session_state:
+    st.session_state['labels_test'] =''
+if 'svm_classifier' not in st.session_state:
+    st.session_state['svm_classifier'] =''
+
+
 st.title("Let's build our Model...")
  
 # Create tabs