Fix example

#1
by Ccellado - opened

Example not working model expects int

label = id2label[str(pred.item())] -> label = id2label[pred.item()]

vLLM Semantic Router org

can you post a PR? Thanks

Sign up or log in to comment