Alex Strick van Linschoten commited on
Commit
e7c875a
1 Parent(s): 39e0016

remove width and height aspect from page add

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -123,9 +123,9 @@ def predict(pdf, confidence, generate_file):
123
  size = img.size
124
  width, height = size
125
  if width > height:
126
- pdf.add_page(orientation="L", format=(height, width))
127
  else:
128
- pdf.add_page(orientation="P", format=(width, height))
129
  pred_dict = model_type.end2end_detect(
130
  img,
131
  valid_tfms,
123
  size = img.size
124
  width, height = size
125
  if width > height:
126
+ pdf.add_page(orientation="L")
127
  else:
128
+ pdf.add_page(orientation="P")
129
  pred_dict = model_type.end2end_detect(
130
  img,
131
  valid_tfms,