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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -35,7 +35,13 @@ photos_folder = "photos"
35
 
36
  # インデックスファイルのパス
37
  index_file_path = "index.csv"
 
 
38
 
 
 
 
 
39
  # タグ付け関数
40
  def predict_tags(image: Image.Image, score_threshold: float) -> tuple[dict[str, float], dict[str, float], str]:
41
  _, height, width, _ = model.input_shape
@@ -84,13 +90,10 @@ def load_data():
84
  return pd.DataFrame(columns=["File Name", "Timestamp", "Tags"])
85
 
86
  # ブラーのトグル用のキー
87
- import random
88
- import string
89
 
90
- blur_toggle_key = "blurOrNot"
91
  #blur_enabled = st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
92
  if "blur_enabled" not in st.session_state:
93
- st.session_state.hoge = st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
94
  # アップロードされた写真を保存する関数
95
  def save_uploaded_photo(uploaded_photo, file_name):
96
  if not os.path.exists(photos_folder):
 
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
 
90
  return pd.DataFrame(columns=["File Name", "Timestamp", "Tags"])
91
 
92
  # ブラーのトグル用のキー
 
 
93
 
 
94
  #blur_enabled = st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
95
  if "blur_enabled" not in st.session_state:
96
+ st.session_state.blur_enabled = st.sidebar.checkbox("NSFW画像にブラーをかける", value=True, key=blur_toggle_key)
97
  # アップロードされた写真を保存する関数
98
  def save_uploaded_photo(uploaded_photo, file_name):
99
  if not os.path.exists(photos_folder):