conhecimento de ferramenta e testes

app.py

@@ -18,9 +18,11 @@ def predict( Fare, Age, Sex, Pclass, SibSp ):
     df = encoding.full_pipeline( df_raw )
 
     model = pkl.load( open( 'models/model_ridge.pkl', 'rb' ) )
-    y_hat  = model._predict_proba_lr( df )[:,1]
+    y_hat  = model._predict_proba_lr( df )[0]
+    
     resp = {'Nada a declarar' : 'NaN'}
-    if y_hat < 0.55:
+
+    if (y_hat[1] < 0.55):
         resp = {'Sobrevivente' : y_hat}
     else:
         resp = {'Não Sobrevivente': y_hat}