hantech commited on
Commit
d2f41b7
Β·
1 Parent(s): 266e7e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def inference(filepath, lang):
55
  config['dataset']['image_min_width'], config['dataset']['image_max_width'])
56
  out = translate(img, model)[0].tolist()
57
  out = vocab.decode(out)
58
- new_bounds.append(bbox, out, prob)
59
  im = PIL.Image.open(img.name)
60
  draw_boxes(im, bounds)
61
  im.save('result.jpg')
 
55
  config['dataset']['image_min_width'], config['dataset']['image_max_width'])
56
  out = translate(img, model)[0].tolist()
57
  out = vocab.decode(out)
58
+ new_bounds.append((bbox, out, prob))
59
  im = PIL.Image.open(img.name)
60
  draw_boxes(im, bounds)
61
  im.save('result.jpg')