diff --git a/notebooks/ai-ml/ImageClassificationInSpark.ipynb b/notebooks/ai-ml/ImageClassificationInSpark.ipynb
index 077247c..430ed72 100644
--- a/notebooks/ai-ml/ImageClassificationInSpark.ipynb
+++ b/notebooks/ai-ml/ImageClassificationInSpark.ipynb
@@ -267,7 +267,7 @@
     "collapsed": false
    },
    "source": [
-    "Create a [Pandas UDF](https://spark.apache.org/docs/3.4.2/api/python/reference/pyspark.sql/api/pyspark.sql.functions.pandas_udf.html) that contains the Torch code that will run on each worker to perform model inference. This training job uses the [ResNet50](https://pytorch.org/vision/main/models/generated/torchvision.models.resnet50.html) model."
+    "Create a [Pandas UDF](https://spark.apache.org/docs/3.4.2/api/python/reference/pyspark.sql/api/pyspark.sql.functions.pandas_udf.html) that contains the Torch code that will run on each worker to perform model inference. This inference job uses the [ResNet50](https://pytorch.org/vision/main/models/generated/torchvision.models.resnet50.html) model."
    ]
   },
   {