From ad94d03576feb3a37a794008118767d1d69d65a3 Mon Sep 17 00:00:00 2001
From: raystlin <raystlin@gmail.com>
Date: Sun, 13 Apr 2025 14:53:34 +0000
Subject: [PATCH] Fixed classify and nsfw tests

They were broken when the constructors changed to include metadata.
---
 internal/ai/classify/model_test.go | 2 +-
 internal/ai/nsfw/nsfw_test.go      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/internal/ai/classify/model_test.go b/internal/ai/classify/model_test.go
index 8483480b8..cab503929 100644
--- a/internal/ai/classify/model_test.go
+++ b/internal/ai/classify/model_test.go
@@ -224,7 +224,7 @@ func TestModel_LoadModel(t *testing.T) {
 		err := tensorFlow.loadModel()
 
 		if err != nil {
-			assert.Contains(t, err.Error(), "no such file or directory")
+			assert.Contains(t, err.Error(), "Could not find SavedModel")
 		}
 
 		assert.Error(t, err)
diff --git a/internal/ai/nsfw/nsfw_test.go b/internal/ai/nsfw/nsfw_test.go
index 59e860b11..41959a050 100644
--- a/internal/ai/nsfw/nsfw_test.go
+++ b/internal/ai/nsfw/nsfw_test.go
@@ -13,7 +13,7 @@ import (
 
 var modelPath, _ = filepath.Abs("../../../assets/nsfw")
 
-var detector = NewModel(modelPath, 224, nil, false)
+var detector = NewModel(modelPath, nil, false)
 
 func TestIsSafe(t *testing.T) {
 	detect := func(filename string) Result {