ivelin commited on
Commit
cdbb363
1 Parent(s): e1fda95

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

Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -51,6 +51,7 @@ def process_refexp(image: Image, prompt: str):
51
  processor.tokenizer.pad_token, "")
52
  # remove first task start token
53
  sequence = re.sub(r"<.*?>", "", sequence, count=1).strip()
 
54
  bbox = processor.token2json(sequence)
55
  print(f"predicted bounding box: {bbox}")
56
 
 
51
  processor.tokenizer.pad_token, "")
52
  # remove first task start token
53
  sequence = re.sub(r"<.*?>", "", sequence, count=1).strip()
54
+ print(fr"predicted decoder sequence before token2json: {sequence}")
55
  bbox = processor.token2json(sequence)
56
  print(f"predicted bounding box: {bbox}")
57