Joyantac33 commited on
Commit
a0166a9
1 Parent(s): df94eec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,12 +13,12 @@ model.to(device)
13
  def process_document(image, question):
14
  # prepare encoder inputs
15
  pixel_values = processor(image, return_tensors="pt").pixel_values
16
- print("pixel_values::........... "+pixel_values)
17
  # prepare decoder inputs
18
  task_prompt = "<s_docvqa><s_question>{user_input}</s_question><s_answer>"
19
  prompt = task_prompt.replace("{user_input}", question)
20
  decoder_input_ids = processor.tokenizer(prompt, add_special_tokens=False, return_tensors="pt").input_ids
21
- print("decoder_input_ids::.................... "+decoder_input_ids)
22
  # generate answer
23
  outputs = model.generate(
24
  pixel_values.to(device),
 
13
  def process_document(image, question):
14
  # prepare encoder inputs
15
  pixel_values = processor(image, return_tensors="pt").pixel_values
16
+ print(pixel_values)
17
  # prepare decoder inputs
18
  task_prompt = "<s_docvqa><s_question>{user_input}</s_question><s_answer>"
19
  prompt = task_prompt.replace("{user_input}", question)
20
  decoder_input_ids = processor.tokenizer(prompt, add_special_tokens=False, return_tensors="pt").input_ids
21
+ print(decoder_input_ids)
22
  # generate answer
23
  outputs = model.generate(
24
  pixel_values.to(device),