Update README.md

README.md

@@ -41,15 +41,17 @@ with torch.inference_mode():
     input_ids = img_proc(img, return_tensors='pt')
     output = model(**input_ids)
 
+threshold=0.4
+
 segmentation_mask = img_proc.post_process_segmentation(
     out_seg,
-    threshold=0.5,
+    threshold=threshold,
     target_sizes=[img.size[::-1]]
 )
 
 bbox_pred = img_proc.post_process_object_detection(
     output,
-    threshold=0.5,
+    threshold=threshold,
     target_sizes=[img.size[::-1]]
 )
 ```