update batch size

app.py

@@ -111,7 +111,7 @@ def retrieval():
             with c2:
                 st.image('figures/multi_molecules.png', use_column_width='always') #, width=125)
             with st.spinner('Loading data...'):
-                batch_size = 64
+                batch_size = 2048
                 dataset = DrugRetrieval(os.path.join(data_path, selected_database), sequence, query_embedding)
                 dataloader = DataLoader(dataset, num_workers=2, batch_size=batch_size, shuffle=False, collate_fn=collate_target)
             st.success('Data loaded.')