fix: typo, related to #15

787d918b · Geewook Kim · GitHub · 4cfb6567 · 787d918b
Unverified Commit 787d918b authored Aug 09, 2022 by Geewook Kim Committed by GitHub Aug 09, 2022
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

test.py test.py +1 -1

No files found.
--- a/test.py
+++ b/test.py
@@ -43,7 +43,7 @@ def test(args):
        if args.task_name == "docvqa":
            output = pretrained_model.inference(
                image=sample["image"],
-                prompt=f"<s_{args.task_name}><s_question>{ground_truth["gt_parses"][0]['question'].lower()}</s_question><s_answer>",
+                prompt=f"<s_{args.task_name}><s_question>{ground_truth['gt_parses'][0]['question'].lower()}</s_question><s_answer>",
            )["predictions"][0]
        else:
            output = pretrained_model.inference(image=sample["image"], prompt=f"<s_{args.task_name}>")["predictions"][0]