hlydecker commited on
Commit
f144b5f
1 Parent(s): f83c136

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from PIL import Image
9
  # TODO: Allow user selectable model?
10
  models = ["model_weights/md_v5a.0.0.pt","model_weights/md_v5b.0.0.pt"]
11
 
12
- def yolo(im, chosen_model, size):
13
  g = (size / max(im.size)) # gain
14
  im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS) # resize
15
 
 
9
  # TODO: Allow user selectable model?
10
  models = ["model_weights/md_v5a.0.0.pt","model_weights/md_v5b.0.0.pt"]
11
 
12
+ def yolo(image, chosen_model, size):
13
  g = (size / max(im.size)) # gain
14
  im = im.resize((int(x * g) for x in im.size), Image.ANTIALIAS) # resize
15