diff --git a/robots-at-robots/research/demos/utils.py b/robots-at-robots/research/demos/utils.py index 93969a53989b036dd763f68b07222bb4522c37a0..6bcbcdba5703dfcfe3dea7e808258dc760555d7f 100644 --- a/robots-at-robots/research/demos/utils.py +++ b/robots-at-robots/research/demos/utils.py @@ -1,9 +1,8 @@ import tensorflow as tf from polystar.common.constants import MODELS_DIR -from polystar.robots_at_robots.globals import settings def load_tf_model(): - model = tf.saved_model.load(export_dir=str(MODELS_DIR / settings.MODEL_NAME / "saved_model")) + model = tf.saved_model.load(export_dir=str(MODELS_DIR / "robots/ssd_mobilenet_v2_coco_2018_03_29" / "saved_model")) return model.signatures["serving_default"]