jinhybr commited on
Commit
3ff8088
1 Parent(s): 9b87379

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "<sroie>"
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