Update streamlit_app.py

streamlit_app.py

@@ -12,9 +12,11 @@ st.markdown(
 
 characters = st.text_input("Enter the characters to begin with")
 
-one_step_reloaded = tf.saved_model.load('reonestep20F')
+epoch = st.radio("Enter the epoch value",['1','5','10','25','50','100','200'])
 
-if characters:
+if characters and epoch:
+
+    one_step_reloaded = tf.saved_model.load(f'model_epoch{epoch}')
 
     states = None
     next_char = tf.constant([characters])