sayedM commited on
Commit
027bb26
1 Parent(s): e9b261e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def yolov8_inference(
37
 
38
  image=image[:, :, ::-1].astype(np.uint8)
39
  model = YOLO("https://huggingface.co/spaces/devisionx/first-demo/blob/main/best_Receipt.pt")
40
- results = model(image,imgsz=320)[0]
41
  image=image[:, :, ::-1].astype(np.uint8)
42
  detections = sv.Detections.from_yolov8(results)
43
  annotated_image = annotatormask.annotate(scene=image, detections=detections)
 
37
 
38
  image=image[:, :, ::-1].astype(np.uint8)
39
  model = YOLO("https://huggingface.co/spaces/devisionx/first-demo/blob/main/best_Receipt.pt")
40
+ results = model(image,imgsz=320,conf=conf_threshold,iou=iou_threshold)[0]
41
  image=image[:, :, ::-1].astype(np.uint8)
42
  detections = sv.Detections.from_yolov8(results)
43
  annotated_image = annotatormask.annotate(scene=image, detections=detections)