From 752a660d2b8938227733413c2dcde460ca83cf3d Mon Sep 17 00:00:00 2001 From: toshiaki1729 <116595002+toshiaki1729@users.noreply.github.com> Date: Tue, 28 May 2024 13:10:35 +0900 Subject: [PATCH] Update README.md --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index c1526a6..45fc89f 100644 --- a/README.md +++ b/README.md @@ -42,8 +42,8 @@ git clone https://github.com/toshiaki1729/stable-diffusion-webui-text2prompt.git It's doing nothing special; - 1. Danbooru tags and it's descriptions are in the `data` folder - -embeddigs of descriptions are generated from wiki + 1. Danbooru tags and it's descriptions are in the `data` folder + - embeddigs of descriptions are generated from wiki - [all-mpnet-base-v2](https://huggingface.co/sentence-transformers/all-mpnet-base-v2) and [all-MiniLM-L6-v2](https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2) models are used to make embeddings from the text 1. Tokenize your input text and calculate cosine similarity with all tag descriptions 1. Choose some tags depending on their similarities