Ankur Goyal commited on
Commit
ab36703
1 Parent(s): 1af0b6d

Increase width

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -5,6 +5,8 @@ os.environ["TOKENIZERS_PARALLELISM"] = "false"
5
  from PIL import ImageDraw
6
  import streamlit as st
7
 
 
 
8
  import torch
9
  from docquery.pipeline import get_pipeline
10
  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:
119
  draw.rectangle(((x1, y1), (x2, y2)), outline=colors[i])
120
 
121
  if document is not None:
122
- col1.image(image, use_column_width=True)
123
 
124
  "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."
125
 
 
5
  from PIL import ImageDraw
6
  import streamlit as st
7
 
8
+ st.set_page_config(layout="wide")
9
+
10
  import torch
11
  from docquery.pipeline import get_pipeline
12
  from docquery.document import load_bytes, load_document
 
121
  draw.rectangle(((x1, y1), (x2, y2)), outline=colors[i])
122
 
123
  if document is not None:
124
+ col1.image(image, use_column_width='auto')
125
 
126
  "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."
127