Update README.md

README.md

@@ -60,9 +60,7 @@ class_queries_logits = outputs.class_queries_logits
 masks_queries_logits = outputs.masks_queries_logits
 
 # you can pass them to processor for postprocessing
-result = processor.post_process_semantic_segmentation(outputs, target_sizes=[image.size[::-1]])[0]
-# we refer to the demo notebooks for visualization (see "Resources" section in the Mask2Former docs)
-predicted_semantic_map = result["segmentation"]
+predicted_semantic_map = processor.post_process_semantic_segmentation(outputs, target_sizes=[image.size[::-1]])[0]
 ```
 
 For more code examples, we refer to the [documentation](https://huggingface.co/docs/transformers/master/en/model_doc/mask2former).