How can I extend this object detection pytorch program to detect and classify more than one class?

Question

I followed this tutorial for building an object detector and now I am trying to expand it to recognise more classes.

This object detection algorithm only works for one cl