Spaces:
Runtime error
Runtime error
ivelin
commited on
Commit
•
72d0321
1
Parent(s):
238e0cb
fix: bugs
Browse filesSigned-off-by: ivelin <ivelin.eth@gmail.com>
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><
|
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
|