Update src/sagemaker_huggingface_inference_toolkit/handler_service.py

Co-authored-by: Philipp Schmid <[email protected]>

Author: vdantu

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.
         """