nehulagrawal commited on
Commit
9542448
1 Parent(s): d89f518

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -33,6 +33,7 @@ def yolov8_inference(
33
  model.overrides['agnostic_nms'] = False # NMS class-agnostic
34
  model.overrides['max_det'] = 1000
35
  image = read_image(image)
 
36
  render = render_result(model=model, image=image, result=results[0])
37
 
38
  return render
 
33
  model.overrides['agnostic_nms'] = False # NMS class-agnostic
34
  model.overrides['max_det'] = 1000
35
  image = read_image(image)
36
+ results = model.predict(image)
37
  render = render_result(model=model, image=image, result=results[0])
38
 
39
  return render