overlayed logits on image. Still having streaming issues cc35efa verified noamholz commited on Jan 21