ivelin commited on
Commit
e1fda95
·
1 Parent(s): 3d33b8d

Signed-off-by: ivelin <[email protected]>

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def process_refexp(image: Image, prompt: str):
46
 
47
  # postprocess
48
  sequence = processor.batch_decode(outputs.sequences)[0]
49
- print(f"predicted decoder sequence: {sequence}")
50
  sequence = sequence.replace(processor.tokenizer.eos_token, "").replace(
51
  processor.tokenizer.pad_token, "")
52
  # remove first task start token
 
46
 
47
  # postprocess
48
  sequence = processor.batch_decode(outputs.sequences)[0]
49
+ print(fr"predicted decoder sequence: {sequence}")
50
  sequence = sequence.replace(processor.tokenizer.eos_token, "").replace(
51
  processor.tokenizer.pad_token, "")
52
  # remove first task start token