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

…-inference Local Classification Inference
roboflow · Dec 22, 2023 · 542d228 · 542d228
2 parents d55f738 + bfa72c3
commit 542d228
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/roboflow/__init__.py b/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):

diff --git a/roboflow/models/classification.py b/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.