salma-remyx commited on
Commit
034eef6
1 Parent(s): 88fac11

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -48,7 +48,7 @@ generated_ids = model.generate(
48
  )
49
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
50
 
51
- parsed_answer = processor.post_process_generation(generated_text, task="<SpatialVQA>", image_size=(image.width, image.height))
52
 
53
  print(parsed_answer)
54
  ```
 
48
  )
49
  generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
50
 
51
+ parsed_answer = processor.post_process_generation(generated_text, task="<POSE>", image_size=(image.width, image.height))
52
 
53
  print(parsed_answer)
54
  ```