Update my_pages/multiverse.py

my_pages/multiverse.py

@@ -282,11 +282,25 @@ def render():
                     y_pred_local = model_train_and_pred(scaler, feature_sel, arch, seed)
                     all_preds.append(y_pred_local)
                 st.markdown(scaler + feature_sel + arch)
-                all_preds_numpy = np.array(all_preds)
-                np.save("all_predictions.npy", all_preds_numpy)
+                # all_preds_numpy = np.array(all_preds)
+                # np.save("all_predictions.npy", all_preds_numpy)
 
     all_preds_numpy = np.array(all_preds)
-    np.save("all_predictions.npy", all_preds_numpy)
+
+    from io import BytesIO
+    
+    # Create a BytesIO object
+    buffer = BytesIO()
+    np.save(buffer, all_preds_numpy)
+    buffer.seek(0)  # move to start
+    
+    st.download_button(
+        label="Download predictions",
+        data=buffer,
+        file_name="all_predictions.npy",
+        mime="application/octet-stream"
+    )
+    # np.save("all_predictions.npy", all_preds_numpy)
     
     uniq_perc = 10.5
     add_red_text(f"""