Update app.py

app.py

@@ -434,13 +434,13 @@ def train_and_evaluate(input_file):
             amaj1=y_train_full.value_counts().idxmax()
             amin1=y_train_full.value_counts().idxmin()
             #print(y.value_counts().idxmax())
-            y_train_full=y_train_full.replace([amin1,amaj1],[1,0])
+            y_train_full=y_train_full.replace([amin1,amaj1],[1,0]).astype(int)  # FIX 1: added .astype(int)
 
             y_test_full = inter_input[outcome_col]
             amaj1=y_test_full.value_counts().idxmax()
             amin1=y_test_full.value_counts().idxmin()
             #print(y.value_counts().idxmax())
-            y_test_full=y_test_full.replace([amin1,amaj1],[1,0])
+            y_test_full=y_test_full.replace([amin1,amaj1],[1,0]).astype(int)  # FIX 1: added .astype(int)
             
             X_train,y_train=X_train_full.values,y_train_full.values
             x_te,y_test=my_test_real.values,y_test_full.values
@@ -591,7 +591,7 @@ def create_interface():
                 metrics_df, calibration_df, calibration_plots = train_and_evaluate(file)
                 
                 if isinstance(metrics_df, str):  # Error case
-                    return metrics_df, None, None, None, None, None, None
+                    return metrics_df, None, None, None, None, None, None, None  # FIX 2: 8 values
                 
 
                 plot_outputs = [None] * 6
@@ -620,4 +620,4 @@ if __name__ == "__main__":
         inbrowser=True,
         height=800,
         show_error=True
-    )
+    )