Merge pull request #210 from roboflow/fix/enable-local-classification-inference

Local Classification Inference

Author: paulguerrie

roboflow/__init__.py

@@ -13,7 +13,7 @@
 from roboflow.models import CLIPModel, GazeModel
 from roboflow.util.general import write_line
 
-__version__ = "1.1.13"
+__version__ = "1.1.14"
 
 
 def check_key(api_key, model, notebook, num_retries=0):

roboflow/models/classification.py

@@ -56,6 +56,10 @@ def __init__(
         self.colors = {} if colors is None else colors
         self.preprocessing = {} if preprocessing is None else preprocessing
 
+        if local is not None:
+            print("initalizing local classification model hosted at :" + local)
+            self.base_url = local
+
     def predict(self, image_path, hosted=False):
         """
         Run inference on an image.