Spaces:

myscale
/

Protein-Structure-Modeling

Runtime error

xuyingli commited on Mar 15, 2023

Commit

9a88d36

1 Parent(s): 8f74fa3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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()
         st.session_state['batch'] = batch_converter
         st.session_state.query_num = 0

     # Initialize SAGE model
     if 'xq' not in st.session_state:
         st.session_state.model, st.session_state.alphabet = init_esm()
+        batch_converter = st.session_state.alphabet.get_batch_converter()
         st.session_state['batch'] = batch_converter
         st.session_state.query_num = 0