Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import tensorflow as tf
 
 st.markdown(
     """
@@ -9,4 +10,16 @@ st.markdown(
     unsafe_allow_html=True
 )
 
-characters = st.text_input("Enter the characters to begin with")
+characters = st.text_input("Enter the characters to begin with")
+
+one_step_reloaded = tf.saved_model.load('reonestep20F')
+
+states = None
+next_char = tf.constant([characters])
+result = [next_char]
+
+for n in range(100):
+  next_char, states = one_step_reloaded.generate_one_step(next_char, states=states)
+  result.append(next_char)
+
+st.write(tf.strings.join(result)[0].numpy().decode("utf-8"))