Alex Strick van Linschoten commited on
Commit
1d9d097
1 Parent(s): 6b5a34f

print pred_dict

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -109,11 +109,13 @@ def predict(pdf, confidence, generate_file):
109
  font_size=16,
110
  label_color="#FF59D6",
111
  )
 
112
  pred_dict["img"].save(
113
  os.path.join(
114
  tmp_dir, filename_without_extension, f"pred-{image}"
115
  )
116
  )
 
117
  pdf.image(
118
  os.path.join(
119
  tmp_dir, filename_without_extension, f"pred-{image}"
109
  font_size=16,
110
  label_color="#FF59D6",
111
  )
112
+ print(pred_dict)
113
  pred_dict["img"].save(
114
  os.path.join(
115
  tmp_dir, filename_without_extension, f"pred-{image}"
116
  )
117
  )
118
+ # TODO: resize image such that it fits the pdf
119
  pdf.image(
120
  os.path.join(
121
  tmp_dir, filename_without_extension, f"pred-{image}"