Upload 6 files

prediction.py

@@ -21,14 +21,13 @@ def get_feeling(number):
 def load_model():
   return tf.keras.models.load_model('model.keras', custom_objects={'KerasLayer': tf_hub.KerasLayer})
 
-if 'the_model' not in st.session_state:
-  with st.spinner("Loading the model, please wait..."):
-    st.session_state.the_model = load_model()
-
 def app():
   st.header('Prediction', divider='rainbow')
 
   user_input = st.text_input("Enter your text here:")
+  if 'the_model' not in st.session_state:
+    with st.spinner("Loading the model, please wait..."):
+      st.session_state.the_model = load_model()
   if st.button('Predict', type="secondary"):
     data = {
       "text_processed": [