package config import ( "path/filepath" tf "github.com/tensorflow/tensorflow/tensorflow/go" ) // TensorFlowVersion returns the TenorFlow framework version. func (c *Config) TensorFlowVersion() string { return tf.Version() } // TensorFlowModelPath returns the TensorFlow model path. func (c *Config) TensorFlowModelPath() string { return filepath.Join(c.AssetsPath(), "nasnet") } // NSFWModelPath returns the "not safe for work" TensorFlow model path. func (c *Config) NSFWModelPath() string { return filepath.Join(c.AssetsPath(), "nsfw") } // FaceNetModelPath returns the FaceNet model path. func (c *Config) FaceNetModelPath() string { return filepath.Join(c.AssetsPath(), "facenet") }