AkashDataScience commited on
Commit
ba0e7f4
·
1 Parent(s): 3012224

Changed argument name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def inference(image, conf_thres, iou_thres,):
18
  pred = model(image, device=device, retina_masks=True, imgsz=1024, conf=conf_thres, iou=iou_thres)
19
  prompt_process = FastSAMPrompt(input, pred, device="cpu")
20
  ann = prompt_process.everything_prompt()
21
- prompt_process.plot(annotations=ann, output="./output.jpg", withContours=False, better_quality=False)
22
  output = Image.open('./output.jpg')
23
  output = np.array(output)
24
  return output
 
18
  pred = model(image, device=device, retina_masks=True, imgsz=1024, conf=conf_thres, iou=iou_thres)
19
  prompt_process = FastSAMPrompt(input, pred, device="cpu")
20
  ann = prompt_process.everything_prompt()
21
+ prompt_process.plot(annotations=ann, output_path="./output.jpg", withContours=False, better_quality=False)
22
  output = Image.open('./output.jpg')
23
  output = np.array(output)
24
  return output