🐛 onnx运行路径问题

internal/application/classifier/classifier.go

@@ -8,6 +8,7 @@ import (
 	"likedream-goframe/internal/framework/logger"
 	"os"
 	"path/filepath"
+	"runtime"
 	"strings"
 
 	ort "github.com/yalue/onnxruntime_go"
@@ -29,6 +30,18 @@ func NewImageClassifier(modelPath, classesPath string) (*ImageClassifier, error)
 	// 获取绝对路径
 	absPath, err := filepath.Abs(exePath)
 	dir := filepath.Dir(absPath)
+	systemOS := runtime.GOOS
+	switch systemOS {
+	case "windows":
+		dir = filepath.Join(dir, "onnxruntime.dll")
+		break
+	case "darwin":
+		dir = filepath.Join(dir, "onnxruntime.dylib")
+		break
+	case "linux":
+		dir = filepath.Join(dir, "onnxruntime.so")
+		break
+	}
 	logger.Logger.Info("初始化onnx：", zap.String("绝对路径", absPath), zap.String("父文件夹", dir))
 	ort.SetSharedLibraryPath(dir)
 	err = ort.InitializeEnvironment()