Update app.py

app.py

@@ -289,7 +289,7 @@ with st.spinner("Loading Models..."):
     # Initialize SAGE model
     if 'xq' not in st.session_state:
         st.session_state.model, st.session_state.alphabet = init_esm()
-        batch_converter = alphabet.get_batch_converter()
+        batch_converter = st.session_state.alphabet.get_batch_converter()
         st.session_state['batch'] = batch_converter
         st.session_state.query_num = 0