Lesterchia174 commited on
Commit
9572155
·
verified ·
1 Parent(s): 1e89571

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def load_model(repo_id):
28
 
29
 
30
  def predict(pilimg):
31
- result = detection_model.predict(pilimg, conf=0.2, iou=0.6)
32
  img_bgr = result[0].plot() # Get image with predictions
33
  out_pilimg = Image.fromarray(img_bgr[..., ::-1]) # Convert BGR to RGB
34
  return out_pilimg
 
28
 
29
 
30
  def predict(pilimg):
31
+ result = detection_model.predict(pilimg, conf=0.5, iou=0.6)
32
  img_bgr = result[0].plot() # Get image with predictions
33
  out_pilimg = Image.fromarray(img_bgr[..., ::-1]) # Convert BGR to RGB
34
  return out_pilimg