Fix the shape error in example

README.md

@@ -44,7 +44,7 @@ IMAGE_PATH = "YOUR_IMAGE_PATH"
 image = Image.open(IMAGE_PATH)
 
 image = processor(image, return_tensors="pt")["pixel_values"]
-image = image.unsqueeze(0).cuda()
+image = image.cuda()
 
 with torch.no_grad():
     (quant_semantic, diff_semantic, indices_semantic, _), \