pass text

app.py

@@ -113,8 +113,8 @@ def process_image(image, task_prompt, text_input=None):
     image = Image.fromarray(image)  # Convert NumPy array to PIL Image
     if task_prompt == 'Document Visual Question Answering':
         task_prompt = '<DocVQA>'
-        results = run_example(task_prompt, image)
-        return results, None
+        results = run_example(task_prompt, image, text_input)
+        return results[task_prompt], None
     elif task_prompt == 'Caption':
         task_prompt = '<CAPTION>'
         results = run_example(task_prompt, image)