Put model serving in its own method

Signed-off-by: Ian Macalinao <me@ian.pw>