From 09760d2b606a1d41b5680f6c74019f0af1ddffcc Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Thu, 10 Nov 2022 09:08:49 +0000 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 4bd7003..c706bff 100644 --- a/README.md +++ b/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). \ No newline at end of file