tokenizer messing smth up

app.py

@@ -53,7 +53,7 @@ def run_example(task_prompt, image, text_input=None):
         task=task_prompt,
         image_size=(image.width, image.height)
     )
-    return parsed_answer
+    return parsed_answer[task_prompt].replace("<pad>", "")
 
 def plot_bbox(image, data):
     fig, ax = plt.subplots()
@@ -114,7 +114,7 @@ def process_image(image, task_prompt, text_input=None):
     if task_prompt == 'Document Visual Question Answering':
         task_prompt = '<DocVQA>'
         results = run_example(task_prompt, image, text_input)
-        return results[task_prompt], None
+        return results, None
     elif task_prompt == 'Caption':
         task_prompt = '<CAPTION>'
         results = run_example(task_prompt, image)