update threshold

ssmmoo1 · ssmmoo1 · commit d8fa78bb7d9d · 2024-01-16T10:37:40.000-06:00
diff --git a/examples/owl_predict.py b/examples/owl_predict.py
@@ -74,7 +74,7 @@
                 image=image, 
                 text=text, 
                 text_encodings=text_encodings,
-                thresholds=thresholds,
+                threshold=thresholds,
                 pad_square=False
             )
         torch.cuda.current_stream().synchronize()

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@`
`74`	`74`	`image=image,`
`75`	`75`	`text=text,`
`76`	`76`	`text_encodings=text_encodings,`
`77`		`- thresholds=thresholds,`
	`77`	`+ threshold=thresholds,`
`78`	`78`	`pad_square=False`
`79`	`79`	`)`
`80`	`80`	`torch.cuda.current_stream().synchronize()`