Spaces:
Runtime error
Runtime error
Joyantac33
commited on
Commit
•
df94eec
1
Parent(s):
2c17cdc
Update app.py
Browse files
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 |
-
|
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 |
-
|
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::........... "+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),
|