Increase width

app.py

@@ -5,6 +5,8 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
 from PIL import ImageDraw
 import streamlit as st
 
+st.set_page_config(layout="wide")
+
 import torch
 from docquery.pipeline import get_pipeline
 from docquery.document import load_bytes, load_document
@@ -119,7 +121,7 @@ if document is not None and question is not None and len(question) > 0:
         draw.rectangle(((x1, y1), (x2, y2)), outline=colors[i])
 
 if document is not None:
-    col1.image(image, use_column_width=True)
+    col1.image(image, use_column_width='auto')
 
 "DocQuery uses LayoutLMv1 fine-tuned on DocVQA, a document visual question answering dataset, as well as SQuAD, which boosts its English-language comprehension. To use it, simply upload an image or PDF, type a question, and click 'submit', or click one of the examples to load them."