Update app.py

app.py

@@ -64,7 +64,7 @@ if submit_button and uploaded_file:
     predictions = pd.read_csv(uploaded_file)
 
     # Check if file format is correct
-    if "patient_id" in predictions.columns and "predicted_survival_time" in predictions.columns:
+    if "patient_id" in predictions.columns and "predicted_scores" in predictions.columns:
         # Merge with ground truth to calculate C-Index
         merged = pd.merge(ground_truth, predictions, on="patient_id", how="inner")
         c_index = concordance_index(event_times=merged["survival_time"], 
@@ -99,7 +99,7 @@ if submit_button and uploaded_file:
         # Display the calculated C-Index to the user
         st.success(f"Submission received! Your C-Index score: {c_index:.4f}")
     else:
-        st.error("Incorrect file format. Ensure columns include 'patient_id' and 'predicted_survival_time'.")
+        st.error("Incorrect file format. Ensure columns include 'patient_id' and 'predicted_scores'.")
 
 # Display the leaderboard
 st.subheader("Leaderboard")