salma-remyx
commited on
Commit
•
034eef6
1
Parent(s):
88fac11
Update README.md
Browse files
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="<
|
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 |
```
|