Update app.py
Browse files
app.py
CHANGED
@@ -35,13 +35,21 @@ photos_folder = "photos"
|
|
35 |
|
36 |
# インデックスファイルのパス
|
37 |
index_file_path = "index.csv"
|
|
|
|
|
|
|
38 |
import random
|
39 |
import string
|
40 |
|
41 |
blur_toggle_key = "blurOrNot"
|
42 |
-
|
|
|
43 |
if "blur_enabled" not in st.session_state:
|
44 |
-
|
|
|
|
|
|
|
|
|
45 |
# タグ付け関数
|
46 |
def predict_tags(image: Image.Image, score_threshold: float) -> tuple[dict[str, float], dict[str, float], str]:
|
47 |
_, height, width, _ = model.input_shape
|
|
|
35 |
|
36 |
# インデックスファイルのパス
|
37 |
index_file_path = "index.csv"
|
38 |
+
|
39 |
+
#blur_enabled = st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
|
40 |
+
# ブラーのトグル用のキー
|
41 |
import random
|
42 |
import string
|
43 |
|
44 |
blur_toggle_key = "blurOrNot"
|
45 |
+
|
46 |
+
# st.session_stateにblur_enabledがない場合に定義
|
47 |
if "blur_enabled" not in st.session_state:
|
48 |
+
st.session_state.blur_enabled = st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
|
49 |
+
|
50 |
+
# blur_enabledの値を取得
|
51 |
+
blur_enabled = st.session_state.blur_enabled
|
52 |
+
|
53 |
# タグ付け関数
|
54 |
def predict_tags(image: Image.Image, score_threshold: float) -> tuple[dict[str, float], dict[str, float], str]:
|
55 |
_, height, width, _ = model.input_shape
|