Skip to content

Commit 8e4d0e3

Browse files
committed
Updated init for classification model to enable local inference
1 parent d55f738 commit 8e4d0e3

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

roboflow/models/classification.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,10 @@ def __init__(
5656
self.colors = {} if colors is None else colors
5757
self.preprocessing = {} if preprocessing is None else preprocessing
5858

59+
if local is not None:
60+
print("initalizing local classification model hosted at :" + local)
61+
self.base_url = local
62+
5963
def predict(self, image_path, hosted=False):
6064
"""
6165
Run inference on an image.

0 commit comments

Comments
 (0)