Added Confidence Argument to keypoint detection model

roboflow/models/keypoint_detection.py

@@ -26,6 +26,7 @@ def __init__(
         id: str,
         name: Optional[str] = None,
         version: Optional[str] = None,
+        confidence: Optional[int] = 10,
         local: Optional[str] = None,
     ):
         """
@@ -37,6 +38,7 @@ def __init__(
             name (str): is the name of the project
             version (str): version number
             local (str): localhost address and port if pointing towards local inference engine
+            confidence (int): A threshold for the returned predictions on a scale of 0-100.
             colors (dict): colors to use for the image
             preprocessing (dict): preprocessing to use for the image
 
@@ -48,6 +50,7 @@ def __init__(
         self.__api_key = api_key
         self.id = id
         self.name = name
+        self.confidence = confidence
         self.version = version
         self.base_url = "https://detect.roboflow.com/"
 
@@ -150,6 +153,7 @@ def __generate_url(self):
                 self.base_url + without_workspace + "/" + str(version),
                 "?api_key=" + self.__api_key,
                 "&name=YOUR_IMAGE.jpg",
+                "&confidence=" + str(self.confidence),
             ]
         )