Update README.md

README.md

@@ -61,10 +61,10 @@ model = AutoModel.from_pretrained(
 
 with torch.inference_mode():
     input_ids = img_proc(img, return_tensors='pt')
-    segmentation = model(**input_ids)
+    output = model(**input_ids)
 
-segmentation_mask = img_proc.post_process_semantic_segmentation(
-    segmentation,
+segmentation = img_proc.post_process_semantic_segmentation(
+    output,
     target_sizes=[img.size[::-1]]
 )
 ```