remove index from downloaded csv

app.py

@@ -155,7 +155,7 @@ def retrieval():
                     mol_img = Chem.Draw.MolToImage(mol)
                     st.image(mol_img, caption=f"{results.loc[j + 5*i, 'Prediction']:.2f}")
 
-        st.download_button('Download retrieved drug compounds.', results.head(selected_k).to_csv().encode('utf-8'), file_name='retrieved_drugs.csv')
+        st.download_button('Download retrieved drug compounds.', results.head(selected_k).to_csv(index=False).encode('utf-8'), file_name='retrieved_drugs.csv')
                     
     elif query_embedding is not None:
         st.markdown('### Inference')
@@ -205,7 +205,7 @@ def retrieval():
                     mol_img = Chem.Draw.MolToImage(mol)
                     st.image(mol_img, caption=f"{results.loc[j + 5*i, 'Prediction']:.2f}")
 
-        st.download_button('Download retrieved drug compounds.', results.head(selected_k).to_csv().encode('utf-8'), file_name='retrieved_drugs.csv')
+        st.download_button('Download retrieved drug compounds.', results.head(selected_k).to_csv(index=False).encode('utf-8'), file_name='retrieved_drugs.csv')
     
 page_names_to_func = {
     'Retrieval': retrieval,