Update app.py
Browse files
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.
|
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):
|