Merge pull request #148 from roboflow/fix/instance-segmentation-local-predict

`local` Parameter For Instance Segmentation Version

Author: paulguerrie

roboflow/__init__.py

@@ -12,7 +12,7 @@
 from roboflow.core.workspace import Workspace
 from roboflow.util.general import write_line
 
-__version__ = "1.0.8"
+__version__ = "1.0.9"
 
 
 def check_key(api_key, model, notebook, num_retries=0):

roboflow/core/version.py

@@ -120,6 +120,7 @@ def __init__(
                     self.id,
                     colors=self.colors,
                     preprocessing=self.preprocessing,
+                    local=local,
                 )
             elif self.type == TYPE_SEMANTIC_SEGMENTATION:
                 self.model = SemanticSegmentationModel(self.__api_key, self.id)

roboflow/models/instance_segmentation.py

@@ -3,13 +3,20 @@
 
 
 class InstanceSegmentationModel(InferenceModel):
-    def __init__(self, api_key, version_id, colors=None, preprocessing=None):
+    def __init__(
+        self, api_key, version_id, colors=None, preprocessing=None, local=None
+    ):
         """
         :param api_key: Your API key (obtained via your workspace API settings page)
         :param version_id: The ID of the dataset version to use for predicting
         """
         super(InstanceSegmentationModel, self).__init__(api_key, version_id)
-        self.api_url = f"{INSTANCE_SEGMENTATION_URL}/{self.dataset_id}/{self.version}"
+        if local is None:
+            self.api_url = (
+                f"{INSTANCE_SEGMENTATION_URL}/{self.dataset_id}/{self.version}"
+            )
+        else:
+            self.api_url = f"{local}/{self.dataset_id}/{self.version}"
         self.colors = {} if colors is None else colors
         self.preprocessing = {} if preprocessing is None else preprocessing