upd

app.py

@@ -45,17 +45,7 @@ def generate_answer_git(image, question):
         "document-question-answering",
         model="andgrt/layoutlmv2-base-uncased_finetuned_docvqa",
     )
-    return qa_pipeline(preprocess_image(image), question)[0]["answer"]
-    # pixel_values, _, _ = preprocess_image(image)
-    # input_ids = processor(text=question, add_special_tokens=False).input_ids
-    # input_ids = [processor.tokenizer.cls_token_id] + input_ids
-    # input_ids = torch.tensor(input_ids).unsqueeze(0).to(device)
-
-    # generated_ids = model.generate(
-    #     pixel_values=pixel_values, input_ids=input_ids, max_length=50
-    # )
-    # generated_answer = processor.batch_decode(generated_ids, skip_special_tokens=True)
-    # return generated_answer[0]
+    return qa_pipeline(image, question)[0]["answer"]
 
 
 def generate_answer(image, question):