Upload app.py

app.py

@@ -78,6 +78,8 @@ def plot_probabilities(probabilities):
         return None
     prob_df = pd.DataFrame({"labels": cell_types, "values": probabilities})
     prob_df.to_csv('Cell_types_predictions.csv', index=False)
+    print("CSV saved!!")
+
     return prob_df
 
 def run_mapping():
@@ -116,7 +118,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='Cell_types_predictions.csv')
 
     nifti_file.change(load_nifti, inputs=nifti_file, outputs=[image_display, slice_slider, gallery, slice_row, plot_row])
     slice_slider.change(update_slice, inputs=slice_slider, outputs=[image_display, prob_plot, gallery])