fix: Removed PDF forced square resize from demo & analysis script (mi…

…ndee#368) * fix: Removed square resize of PDF in analysis script * fix: Removed forced-square resize from demo
ANASS812 · Jul 7, 2021 · 194d7c6 · 194d7c6
1 parent a73a9f3
commit 194d7c6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/demo/app.py b/demo/app.py
@@ -47,7 +47,7 @@ def main():
     uploaded_file = st.sidebar.file_uploader("Upload files", type=['pdf', 'png', 'jpeg', 'jpg'])
     if uploaded_file is not None:
         if uploaded_file.name.endswith('.pdf'):
-            doc = DocumentFile.from_pdf(uploaded_file.read()).as_images(output_size=(1024, 1024))
+            doc = DocumentFile.from_pdf(uploaded_file.read()).as_images()
         else:
             doc = DocumentFile.from_images(uploaded_file.read())
         cols[0].image(doc[0], width=640)

diff --git a/scripts/analyze.py b/scripts/analyze.py
@@ -22,7 +22,7 @@ def main(args):
 
     model = ocr_predictor(args.detection, args.recognition, pretrained=True)
     if args.path.endswith(".pdf"):
-        doc = DocumentFile.from_pdf(args.path).as_images(output_size=(1024, 1024))
+        doc = DocumentFile.from_pdf(args.path).as_images()
     else:
         doc = DocumentFile.from_images(args.path)