chunk_content: o_grad():
      outputs = inference_model(pixel_values=encoding.pixel_values)
      logits = outputs.logits
  
  upsampled_logits = nn.functional.interpolate(
      logits,
      size=image.size[::-1],
      mode="bilinear",
      align_corners=False,
  )
  
  pred_seg = upsampled_logits.argmax(dim=1)[0]
  Next, visualize the results.  We need a color palette for this. Here, we use ade_palette(). As it is a long array, so
  we don’t include it in this guide, please copy