Merge pull request #146 from roboflow/use-propper-env-vars-for-inference-endpoints

use OBJECT_DETECTION_URL config value rather than hardcoded prod url for inference

Author: hansent

roboflow/models/object_detection.py

@@ -15,7 +15,7 @@
 import wget
 from PIL import Image
 
-from roboflow.config import API_URL, OBJECT_DETECTION_MODEL
+from roboflow.config import API_URL, OBJECT_DETECTION_MODEL, OBJECT_DETECTION_URL
 from roboflow.util.image_utils import check_image_url
 from roboflow.util.prediction import PredictionGroup
 from roboflow.util.versions import (
@@ -80,7 +80,7 @@ def __init__(
 
         # local needs to be passed from Project
         if local is None:
-            self.base_url = "https://detect.roboflow.com/"
+            self.base_url = OBJECT_DETECTION_URL + "/"
         else:
             print("initalizing local object detection model hosted at :" + local)
             self.base_url = local
@@ -515,7 +515,7 @@ def __generate_url(
         # Reassign parameters if any parameters are changed
         if local is not None:
             if not local:
-                self.base_url = "https://detect.roboflow.com/"
+                self.base_url = OBJECT_DETECTION_URL + "/"
             else:
                 self.base_url = "http://localhost:9001/"