Update app.py

app.py

@@ -64,9 +64,9 @@ if txt is not None:
         with col1:
             options = st.multiselect(
             'Which image you want download?',
-            [1, 2, 3, 4, 5])
+            [i in range(top_k)])
 
-            img_for_download = [list_image[int(i)-1] for i in options]
+            img_for_download = [list_images[int(i)-1] for i in options]
             st.write('You selected:', img_for_download)
             
             # btn = st.download_button(