Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def process_document(image):
|
|
15 |
pixel_values = processor(image, return_tensors="pt").pixel_values
|
16 |
|
17 |
# prepare decoder inputs
|
18 |
-
task_prompt = "<
|
19 |
decoder_input_ids = processor.tokenizer(task_prompt, add_special_tokens=False, return_tensors="pt").input_ids
|
20 |
|
21 |
# generate answer
|
|
|
15 |
pixel_values = processor(image, return_tensors="pt").pixel_values
|
16 |
|
17 |
# prepare decoder inputs
|
18 |
+
task_prompt = "<s_cord-v2>"
|
19 |
decoder_input_ids = processor.tokenizer(task_prompt, add_special_tokens=False, return_tensors="pt").input_ids
|
20 |
|
21 |
# generate answer
|