diff --git a/examples/alexnet/alexnet.py b/examples/alexnet/alexnet.py
index de401d2..63bb580 100644
--- a/examples/alexnet/alexnet.py
+++ b/examples/alexnet/alexnet.py
@@ -34,5 +34,5 @@ if __name__ == "__main__":
     alexnet = torchvision.models.alexnet(weights=torchvision.models.AlexNet_Weights.DEFAULT)
     print_model_parameters(alexnet)  # print layer names and number of parameters
     export_model_weights(alexnet, 'alexnet_weights.bin')
-    # predict('cat.jpg')
+    # predict(alexnet, 'cat.jpg')