Update README.md

README.md

@@ -46,7 +46,7 @@ Here is how to use this model:
 >>> masks_queries_logits = outputs.masks_queries_logits
 
 >>> # you can pass them to feature_extractor for postprocessing
->>> output = feature_extractor.post_process_semantic_segmentation(outputs)
+>>> predicted_semantic_map = feature_extractor.post_process_semantic_segmentation(outputs)[0]
 ```
 
 For more code examples, we refer to the [documentation](https://huggingface.co/docs/transformers/master/en/model_doc/maskformer).