Improve answer preview

app.py

@@ -126,10 +126,13 @@ def process_question(question, document, model=list(CHECKPOINTS.keys())[0]):
     if document is None:
         return None, None, None
 
+    text_value = None
     predictions = run_pipeline(model, question, document, 3)
     pages = [x.copy().convert("RGB") for x in document.preview]
     for i, p in enumerate(ensure_list(predictions)):
-        if i > 0:
+        if i == 0:
+            text_value = p["answer"]
+        else:
             # Keep the code around to produce multiple boxes, but only show the top
             # prediction for now
             break
@@ -151,7 +154,7 @@ def process_question(question, document, model=list(CHECKPOINTS.keys())[0]):
         gr.update(visible=True, value=predictions),
         gr.update(
             visible=True,
-            value=predictions[0]["answer"] if ensure_list(predictions) else None,
+            value=text_value,
         ),
     )