JoanaS commited on
Commit
88c3302
·
verified ·
1 Parent(s): 02c95e0

Update app.py

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