Hugging Face's logo

Spaces:
HarishMaths
/
CharPred
Sleeping

App Files Community
HarishMaths commited on
Commit
3c8a9bb
·
verified ·
1 Parent(s): 11e9a18

Update streamlit_app.py

Browse files
Files changed (1) hide show
  1. streamlit_app.py +1 -1
streamlit_app.py CHANGED
@@ -16,7 +16,7 @@ epoch = st.radio("Enter the epoch value",['1','5','10','25','50','100','200'])
16
 
17
  if characters and epoch:
18
 
19
- one_step_reloaded = tf.saved_model.load('dummy')
20
 
21
  states = None
22
  next_char = tf.constant([characters])
 
16
 
17
  if characters and epoch:
18
 
19
+ one_step_reloaded = tf.saved_model.load(f'model_epoch{epoch}')
20
 
21
  states = None
22
  next_char = tf.constant([characters])