Update src/sagemaker_huggingface_inference_toolkit/handler_service.py

Co-authored-by: Philipp Schmid <32632186+philschmid@users.noreply.github.com>

src/sagemaker_huggingface_inference_toolkit/handler_service.py

@@ -133,7 +133,7 @@ def predict(self, data, model):
         The predict handler can be overridden to implement the model inference.
         Args:
             data (dict): deserialized decoded_input_data returned by the input_fn
-            model : Model returned by "model_fn".
+            model : Model returned by the `load` method or if it is a custom module `model_fn`.
         Returns:
             obj (dict): prediction result.
         """