Fix images

app.py

@@ -10,6 +10,7 @@ import gradio as gr
 import torch
 from docquery.pipeline import get_pipeline
 from docquery.document import load_bytes, load_document, ImageDocument
+from docquery.ocr_reader import get_ocr_reader
 
 
 def ensure_list(x):
@@ -173,10 +174,9 @@ def process_question(question, document, model=list(CHECKPOINTS.keys())[0]):
 def load_example_document(img, question, model):
     if img is not None:
         if question in question_files:
-            print("using document")
             document = load_document(question_files[question])
         else:
-            document = ImageDocument(Image.fromarray(img))
+            document = ImageDocument(Image.fromarray(img), get_ocr_reader())
         preview, answer, answer_text = process_question(question, document, model)
         return document, question, preview, gr.update(visible=True), answer, answer_text
     else: