ydshieh HF staff commited on
Commit
a6e997e
1 Parent(s): 496517f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -49,13 +49,13 @@ generated_ids = model.generate(
49
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
50
 
51
  # Specify `cleanup_and_extract=False` in order to see the raw model generation.
52
- processed_text = processor.post_processor_generation(generated_text, cleanup_and_extract=False)
53
 
54
  print(processed_text)
55
  # `<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>.`
56
 
57
  # By default, the generated text is cleanup and the entities are extracted.
58
- processed_text, entities = processor.post_processor_generation(generated_text)
59
 
60
  print(processed_text)
61
  # `An image of a snowman warming himself by a fire.`
 
49
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
50
 
51
  # Specify `cleanup_and_extract=False` in order to see the raw model generation.
52
+ processed_text = processor.post_process_generation(generated_text, cleanup_and_extract=False)
53
 
54
  print(processed_text)
55
  # `<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>.`
56
 
57
  # By default, the generated text is cleanup and the entities are extracted.
58
+ processed_text, entities = processor.post_process_generation(generated_text)
59
 
60
  print(processed_text)
61
  # `An image of a snowman warming himself by a fire.`