Update app.py

app.py

@@ -212,14 +212,15 @@ with col_2:
     # If file is uploaded, read in the data and plot it
     
 if uploaded_file is not None:
-    data, wcs = load_file(uploaded_file)
-    os.system(f'mkdir -p {uploaded_file.name.strip(".fits")}')
     fname = uploaded_file.name.strip(".fits")
+    os.system(f'mkdir -p {fname}')
+    data, wcs = load_file(uploaded_file)
     st.session_state['loaded'] = True
     st.session_state['example'] = False
 
 if example:
     fname = "NGC4649_example"
+    os.system(f'mkdir -p {fname}')
     data, wcs = load_file(f"{fname}.fits")
     st.session_state['loaded'] = False
     st.session_state['example'] = True