Update app.py

app.py

@@ -196,7 +196,8 @@ if uploaded_file is not None:
     plot_decomposed(image_decomposed)
 
     with col6:
-        pass
+        with open('predicted.zip', 'rb') as f:
+            res = f.read()
         # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
     #     # download = st.button('Download')
         download = st.download_button(label="Download", data=res, file_name="predicted.zip", mime="application/octet-stream")