From 9fcffec0e62121146b65e2ecbb2ebcab7eae72c2 Mon Sep 17 00:00:00 2001
From: Abhinav <abhinavk.grd@gmail.com>
Date: Tue, 22 Feb 2022 08:17:50 +0530
Subject: [PATCH] pass modelURL while loading SSDMobileNet to load the custom
 model

---
 src/services/machineLearning/ssdMobileNetV2Service.ts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/services/machineLearning/ssdMobileNetV2Service.ts b/src/services/machineLearning/ssdMobileNetV2Service.ts
index ad8715c27..eb3aca0db 100644
--- a/src/services/machineLearning/ssdMobileNetV2Service.ts
+++ b/src/services/machineLearning/ssdMobileNetV2Service.ts
@@ -29,6 +29,7 @@ class SSDMobileNetV2 implements ObjectDetectionService {
         this.ssdMobileNetV2Model = await SSDMobileNet.load({
             base: 'mobilenet_v2',
             dataset: 'open_images',
+            modelUrl: '/models/open-images-ssd-mobilenet-v2/model.json',
         });
         console.log(
             'loaded ssdMobileNetV2Model',