update figure references

app.py

@@ -104,7 +104,7 @@ def retrieval():
                 'Select target encoder',(['SeqVec'])
             )
         if sequence:
-            st.image('figures/protein_encoder_done.png', use_column_width='always')
+            st.image('figures/target_encoder_done.png', use_column_width='always')
             with st.spinner('Encoding in progress...'):
 
                 with open(os.path.join(data_path, f'Lenselink/processed/SeqVec_encoding_test.pickle'), 'rb') as handle:
@@ -130,7 +130,7 @@ def retrieval():
             selected_database = 'DUDE'
         c1, c2 = st.columns(2)
         with c2:
-            st.image('figures/multi_molecules.png', use_column_width='always') #, width=125)
+            st.image('figures/multi_drugs.png', use_column_width='always') #, width=125)
         with st.spinner('Loading data...'):
             batch_size = 2048
             dataset = DrugRetrieval(os.path.join(data_path, selected_database), sequence, query_embedding)
@@ -141,7 +141,7 @@ def retrieval():
         selected_encoder = st.selectbox(
                 'Select drug encoder',(['CDDD'])
             )
-        st.image('figures/molecule_encoder_done.png', use_column_width='always')
+        st.image('figures/drug_encoder_done.png', use_column_width='always')
         st.success('Encoding complete.')
                 
     if sequence == ex_target and selected_database == 'Lenselink':