Incorrect confidence type annotation

autodistill · Feb 20, 2024 · a246dcc · a246dcc
1 parent c1d9e6c
commit a246dcc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autodistill_owlv2/owlv2_model.py b/autodistill_owlv2/owlv2_model.py
@@ -37,7 +37,7 @@ def __init__(self, ontology: CaptionOntology):
         )
         self.ontology = ontology
 
-    def predict(self, input: Any, confidence: int = 0.1) -> sv.Detections:
+    def predict(self, input: Any, confidence: float = 0.1) -> sv.Detections:
         image = load_image(input, return_format="PIL")
         texts = [self.ontology.prompts()]