Cyrile commited on
Commit
eefdbff
1 Parent(s): 2e95895

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -68,7 +68,7 @@ with torch.inference_mode():
68
  threshold=0.4
69
 
70
  segmentation_mask = img_proc.post_process_segmentation(
71
- out_seg,
72
  threshold=threshold,
73
  target_sizes=[img.size[::-1]]
74
  )
 
68
  threshold=0.4
69
 
70
  segmentation_mask = img_proc.post_process_segmentation(
71
+ output,
72
  threshold=threshold,
73
  target_sizes=[img.size[::-1]]
74
  )