tagger

Documentation

Index

• type Client
  • func NewClient(baseUrl string) *Client
  • func (c *Client) GetInfo() (*ServiceInfo, error)
  • func (c *Client) GetTaggerState() (*TaggerState, error)
  • func (c *Client) SwitchModel(name string) error
  • func (c *Client) TagImage(reader io.Reader, taggerModel string) ([]ImageTag, error)
• type ImageTag
• type ImageTaggerPlugin
  • func NewImageTaggerPlugin() *ImageTaggerPlugin
  • func (i *ImageTaggerPlugin) IsEnable() bool
  • func (i *ImageTaggerPlugin) OnInit(e *harukap.HarukaAppEngine) error
• type ResponseWrap
• type ServiceInfo
• type TaggerState

Types

type Client

type Client struct {
	BaseUrl string
	// contains filtered or unexported fields
}

func NewClient

func NewClient(baseUrl string) *Client

func (*Client) GetInfo

func (c *Client) GetInfo() (*ServiceInfo, error)

func (*Client) GetTaggerState

func (c *Client) GetTaggerState() (*TaggerState, error)

func (*Client) SwitchModel

func (c *Client) SwitchModel(name string) error

func (*Client) TagImage

func (c *Client) TagImage(reader io.Reader, taggerModel string) ([]ImageTag, error)

type ImageTag

type ImageTag struct {
	Tag  string  `json:"tag"`
	Rank float64 `json:"rank"`
}

type ImageTaggerPlugin

type ImageTaggerPlugin struct {
	Client *Client
	Enable bool
}

func NewImageTaggerPlugin

func NewImageTaggerPlugin() *ImageTaggerPlugin

func (*ImageTaggerPlugin) IsEnable

func (i *ImageTaggerPlugin) IsEnable() bool

func (*ImageTaggerPlugin) OnInit

func (i *ImageTaggerPlugin) OnInit(e *harukap.HarukaAppEngine) error

type ResponseWrap

type ResponseWrap[T interface{}] struct {
	Success bool   `json:"success"`
	Error   string `json:"error"`
	Data    T      `json:"data"`
}

type ServiceInfo

type ServiceInfo struct {
	Name    string `json:"name"`
	Success bool   `json:"success"`
}

type TaggerState

type TaggerState struct {
	ModelName string   `json:"modelName"`
	ModelList []string `json:"modelList"`
}