Update app.py

app.py

@@ -95,12 +95,12 @@ outputs = [gr.Dataframe(
 def infer(inputs):
     data = pd.DataFrame(inputs, columns=headers)
     
+    # Store the Name column first, before any processing
+    names = data['Name'].copy()
+    
     # Replace empty strings with NaN
     data = data.replace('', np.nan)
     
-    # Store the Name column
-    names = data['Name'].copy()
-    
     # Add missing columns with default values
     for col in all_headers:
         if col not in data.columns:
@@ -119,7 +119,7 @@ def infer(inputs):
     # Make predictions
     predictions = pipe.predict(data)
     
-    # Create output DataFrame with original names and predictions
+    # Create output DataFrame with stored names and predictions
     return pd.DataFrame({
         'Name': names,
         'Depression': predictions
@@ -127,6 +127,7 @@ def infer(inputs):
 
 
 
+
 gr.Interface(
     fn=infer,
     inputs=inputs,