ivelin commited on
Commit
72d0321
1 Parent(s): 238e0cb

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

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def process_refexp(image: Image, prompt: str):
26
  pixel_values = processor(image, return_tensors="pt").pixel_values
27
 
28
  # prepare decoder inputs
29
- task_prompt = "<s_refexp><s_prompt>{user_input}</s_prompt><s_refexp>"
30
  prompt = task_prompt.replace("{user_input}", prompt)
31
  decoder_input_ids = processor.tokenizer(
32
  prompt, add_special_tokens=False, return_tensors="pt").input_ids
 
26
  pixel_values = processor(image, return_tensors="pt").pixel_values
27
 
28
  # prepare decoder inputs
29
+ task_prompt = "<s_refexp><s_prompt>{user_input}</s_prompt><s_target_bounding_box>"
30
  prompt = task_prompt.replace("{user_input}", prompt)
31
  decoder_input_ids = processor.tokenizer(
32
  prompt, add_special_tokens=False, return_tensors="pt").input_ids