Update README.md

README.md

@@ -48,7 +48,7 @@ generated_ids = model.generate(
 )
 generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
 
-parsed_answer = processor.post_process_generation(generated_text, task="<POSE>", image_size=(image.width, image.height))
+parsed_answer = processor.post_process_generation(generated_text, task=prompt, image_size=(image.width, image.height))
 
 print(parsed_answer)
 ```