OzoneAsai commited on
Commit
8ca8419
1 Parent(s): 8a5b81f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
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
- #blur_enabled = st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
 
43
  if "blur_enabled" not in st.session_state:
44
- st.session_state.blur_enabled = st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
 
 
 
 
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