xuyingli commited on
Commit
9716a19
1 Parent(s): 6fb8953

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -291,6 +291,7 @@ with st.spinner("Loading Models..."):
291
  st.session_state.model, st.session_state.alphabet = init_esm()
292
  batch_converter = st.session_state.alphabet.get_batch_converter()
293
  st.session_state['batch'] = batch_converter
 
294
  st.session_state.query_num = 0
295
 
296
  if 'xq' not in st.session_state:
 
291
  st.session_state.model, st.session_state.alphabet = init_esm()
292
  batch_converter = st.session_state.alphabet.get_batch_converter()
293
  st.session_state['batch'] = batch_converter
294
+ st.session_state.batch_converter = st.session_state.alphabet.get_batch_converter()
295
  st.session_state.query_num = 0
296
 
297
  if 'xq' not in st.session_state: