Update app.py

app.py

@@ -411,7 +411,10 @@ else:
             
             if sequence:
                 st.write('you have entered: ', sequence)
-                result_temp_seq = esm_search(st.session_state['xq'], sequence, st.session_state['batch'] ,top_k=10)
+                st.session_state['xq'] = model
+                batch_converter = alphabet.get_batch_converter()
+                
+                result_temp_seq = esm_search(st.session_state['xq'], sequence, batch_converter ,top_k=10)
                 st.text('search result (top 5): ')
                 # tab1, tab2, tab3, tab4, = st.tabs(["Cat", "Dog", "Owl"])
                 tab1, tab2, tab3 , tab4, tab5 = st.tabs(['1','2','3','4','5'])