Update app.py

app.py

@@ -46,7 +46,6 @@ df = pd.read_csv('results.csv',
                  names = ['image_name', 'comment_number', 'comment'], 
                  header=0)
 
-
 if txt is not None:
      if st.button('Find!'):
         list_images, img_desc, list_images_vit, img_descs_vit = find_image_disc(txt, df, top_k)
@@ -59,4 +58,12 @@ if txt is not None:
                 st.write(img_desc[ind])
             with col2:
                 st.image(pic[1])
-                st.write(img_descs_vit[ind])
+                st.write(img_descs_vit[ind])
+
+        with col1:
+            btn = st.download_button(
+            label="Download all images by ResNet50",
+            data=list_images,
+            file_name=list_images,
+            mime="image/png"
+            )