Update app.py

app.py

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