Hi Cristelle,
can you post the code you used?
Using the beginner colab 5, after the cell with model_1.save(…) (in the section Prepare this model for TensorFlow Serving) I’ve tried this:
for r in test_ds.take(1):
  print(model_1(r[0]))

And I got the expected predictions