Rename method

README.md

@@ -49,13 +49,13 @@ generated_ids = model.generate(
 generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
 # Specify `cleanup_and_extract=False` in order to see the raw model generation.
-processed_text = processor.post_processor_generation(generated_text, cleanup_and_extract=False)
+processed_text = processor.post_process_generation(generated_text, cleanup_and_extract=False)
 
 print(processed_text)
 # `<grounding> An image of<phrase> a snowman</phrase><object><patch_index_0044><patch_index_0863></object> warming himself by<phrase> a fire</phrase><object><patch_index_0005><patch_index_0911></object>.`
 
 # By default, the generated  text is cleanup and the entities are extracted.
-processed_text, entities = processor.post_processor_generation(generated_text)
+processed_text, entities = processor.post_process_generation(generated_text)
 
 print(processed_text)
 # `An image of a snowman warming himself by a fire.`

processing_kosmos2.py

@@ -336,7 +336,7 @@ class Kosmos2Processor(ProcessorMixin):
         """
         return self.tokenizer.decode(*args, **kwargs)
 
-    def post_processor_generation(self, text, cleanup_and_extract=True):
+    def post_process_generation(self, text, cleanup_and_extract=True):
 
         caption = text.split("</image>")[-1]
         if cleanup_and_extract: