diff --git a/src/litserve/loops/base.py b/src/litserve/loops/base.py index 6c366b0a..89776377 100644 --- a/src/litserve/loops/base.py +++ b/src/litserve/loops/base.py @@ -283,13 +283,13 @@ def pre_setup(self, lit_api: LitAPI, spec: Optional[LitSpec]): """When `stream=False`, `lit_api.predict`, `lit_api.encode_response` must not be generator functions. - Correct example: + Correct usage: def predict(self, inputs): ... return {"output": output} - Incorrect example: + Incorrect usage: def predict(self, inputs): ...