ivelin commited on
Commit
949256f
1 Parent(s): 9184635

Signed-off-by: ivelin <ivelin.eth@gmail.com>

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -55,6 +55,7 @@ def process_refexp(image: Image, prompt: str):
55
  print(
56
  fr"predicted decoder sequence before token2json: {html.escape(sequence)}")
57
  bbox = processor.token2json(sequence)
 
58
  print(f"predicted bounding box: {bbox}")
59
 
60
  print(f"image object: {image}")
 
55
  print(
56
  fr"predicted decoder sequence before token2json: {html.escape(sequence)}")
57
  bbox = processor.token2json(sequence)
58
+ bbox = bbox['target_bounding_box']
59
  print(f"predicted bounding box: {bbox}")
60
 
61
  print(f"image object: {image}")