Alex Strick van Linschoten commited on
Commit
f5429b9
·
1 Parent(s): 404174f

fix aspect issues

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((height, width))
127
  else:
128
- pdf.add_page((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("L", (height, width))
127
  else:
128
+ pdf.add_page("P", (width, height))
129
  pred_dict = model_type.end2end_detect(
130
  img,
131
  valid_tfms,