Prediction.ipynb errors

[amullapu]

I tried to run the inferenceconfig function and I get the following error

NameError Traceback (most recent call last)
in ()
----> 1 class InferenceConfig(surgery.SurgeryConfig):
2 # Set batch size to 1 since we'll be running inference on
3 # one image at a time. Batch size = GPU_COUNT * IMAGES_PER_GPU
4 GPU_COUNT = 1
5 IMAGES_PER_GPU = 1

NameError: name 'surgery' is not defined