# Demo usage for inference or loading pretrained model