If anyone is interested: There’s a SWAG notebook in community section notebook that can be used for reference. The input can be context: sentence. 1: option_1, 2: option_2 where labels can be option_numbers.