minor update

app.py

@@ -432,7 +432,8 @@ def plot_probabilities(probabilities):
     if len(probabilities) < 1:
         return None
     prob_df = pd.DataFrame({"labels": cell_types, "values": probabilities})
-    prob_df.to_csv('./Cell_types_predictions.csv', index=False)
+    os.makedirs("outputs", exist_ok=True)
+    prob_df.to_csv('outputs/Cell_types_predictions.csv', index=False)
     return prob_df
 
 def run_mapping():
@@ -456,7 +457,7 @@ def run_registration(data_type, selected_idx):
     return "Registration complete!"
 
 def download_csv():
-    return './Cell_types_predictions.csv'
+    return 'outputs/Cell_types_predictions.csv'
 
 
 def handle_data_type_change(dt):
@@ -498,7 +499,7 @@ with gr.Blocks() as demo:
         gr.Markdown("### Step 5: Quantitative results of the mapping model.")
         prob_plot = gr.BarPlot(prob_df, x="labels", y="values", title="Cell Type Probabilities", scroll_to_output=True, x_label_angle=-90, height=400)
         gr.Markdown("### Step 6: Download Results.")
-        download_button = gr.DownloadButton(label="Download Results", value='./Cell_types_predictions.csv')
+        download_button = gr.DownloadButton(label="Download Results", value='outputs/Cell_types_predictions.csv')
 
     nifti_file.change(
     load_nifti_or_png,