Skip to content

Commit 4d7829a

Browse files
Update README.md
1 parent de785b3 commit 4d7829a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ This toy example with 2 models (inference pipeline) shows LitServe's flexibility
7373
import litserve as ls
7474

7575
# define the api to include any number of models, dbs, etc...
76-
class SimpleLitAPI(ls.LitAPI):
76+
class InferencePipeline(ls.LitAPI):
7777
def setup(self, device):
7878
self.model1 = lambda x: x**2
7979
self.model2 = lambda x: x**3
@@ -95,7 +95,7 @@ class SimpleLitAPI(ls.LitAPI):
9595

9696
if __name__ == "__main__":
9797
# 12+ features like batching, streaming, etc...
98-
server = ls.LitServer(SimpleLitAPI(max_batch_size=1), accelerator="auto")
98+
server = ls.LitServer(InferencePipeline(max_batch_size=1), accelerator="auto")
9999
server.run(port=8000)
100100
```
101101

0 commit comments

Comments
 (0)