Update app.py

app.py

@@ -161,7 +161,8 @@ with col:
     
 # If file is uploaded, read in the data and plot it
 if uploaded_file is not None:
-    fname = uploaded_file.strip(".fits")
+    # fname = uploaded_file.strip(".fits")
+    uploaded_file
     
     with fits.open(uploaded_file) as hdul:
         data = hdul[0].data
@@ -236,4 +237,4 @@ if uploaded_file is not None:
                 st.markdown("")
                 st.markdown("")
                 # st.markdown("""<style>[data-baseweb="select"] {margin-top: 16px;}</style>""", unsafe_allow_html=True)
-                download = st.download_button(label="Download", data=res, file_name=f'{fname}_{int(scale*128)}.zip', mime="application/octet-stream")
+                download = st.download_button(label="Download", data=res, file_name=f'pred_{int(scale*128)}.zip', mime="application/octet-stream")