Fixing detected issue in gs guide

Author: davidrsch

vignettes/getting_started.Rmd

@@ -214,15 +214,15 @@ Once the model is trained, we can use the standard `tidymodels` `predict()` func
 ```{r model-predict-class}
 # Predict the class for the first 5 images in the test set 
 class_preds <- mlp_fit |>
-  predict(new_data = head(test_df))
+  predict(new_data = head(select(test_df, x)))
 class_preds
 ```
 
 To get the underlying probabilities for each class, we can set `type = "prob"`. This returns a tibble with a probability column for each of the 10 classes (0-9).
 
 ```{r model-predict-prob}
 # Predict probabilities for the first 5 images
-prob_preds <- mlp_fit |> predict(new_data = head(test_df), type = "prob")
+prob_preds <- mlp_fit |> predict(new_data = head(select(test_df, x)), type = "prob")
 prob_preds
 ```