Update app.py

app.py

@@ -86,7 +86,11 @@ def infer(inputs):
     data = data.astype(float)
     
     predictions = pipe.predict(data)
-    return pd.DataFrame(predictions, columns=["Name", "Depression"])
+    #return pd.DataFrame(predictions, columns=["Name", "Depression"])
+    return pd.DataFrame({
+        'Name': data['Name'],
+        'Depression': predictions
+    })