fix: build

Stormix · Nov 12, 2023 · 10816d8 · 10816d8
1 parent bf491f5
commit 10816d8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/client/src/lib/nsfw.ts b/client/src/lib/nsfw.ts
@@ -137,9 +137,9 @@ export class NSFWJS {
 
       let model: tf.LayersModel | tf.GraphModel
       if (endpoint == null) {
-        model = this.model
+        model = this.model!
       } else {
-        if (this.model.hasOwnProperty('layers') && this.intermediateModels[endpoint] == null) {
+        if (this.model!.hasOwnProperty('layers') && this.intermediateModels[endpoint] == null) {
           // @ts-ignore
           const layer = this.model.layers.find((l) => l.name === endpoint)
           this.intermediateModels[endpoint] = tf.model({