Update app.py

app.py

@@ -16,22 +16,6 @@ import numpy as np
 # ページごとの表示数
 PAGE_SIZE = 20
 
-# Streamlit アプリケーションのタイトル
-st.title("NSFW ブラーフォトアップロードとタグ付けアプリ")
-
-# フォトフォルダーが存在しない場合は作成
-photos_folder = "photos"
-if not os.path.exists(photos_folder):
-    os.makedirs(photos_folder)
-
-# アップロードされた写真の情報を格納するデータフレーム
-photo_df = pd.DataFrame(columns=["File Name", "Timestamp", "Tags"])
-
-# インデックスのCSVファイルが存在しない場合は作成
-index_file_path = "index.csv"
-if not os.path.exists(index_file_path):
-    pd.DataFrame(columns=["File Name", "Timestamp", "Tags"]).to_csv(index_file_path, index=False)
-
 # タグ付け関数
 def predict_tags(image: Image.Image, score_threshold: float) -> tuple[dict[str, float], dict[str, float], str]:
     _, height, width, _ = model.input_shape
@@ -90,15 +74,15 @@ def download_photos_as_zip(file_paths):
             unsafe_allow_html=True
         )
 
-# モデルとラベルをダウンロードする
-model = load_model()
-labels = load_labels()
-
 # Streamlit アプリケーションのメイン部分
 def main():
     st.sidebar.title("アップロードオプション")
     uploaded_photos = st.sidebar.file_uploader("写真をアップロードしてください", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
 
+    # モデルとラベルをダウンロードする
+    model = load_model()
+    labels = load_labels()
+
     if uploaded_photos:
         for uploaded_photo in uploaded_photos:
             file_name = f"{datetime.now().strftime('%Y%m%d%H%M%S%f')}.png"
@@ -155,4 +139,7 @@ def main():
         download_photos_as_zip(file_paths)
 
 if __name__ == "__main__":
+    # モデルとラベルをダウンロードする
+    model = load_model()
+    labels = load_labels()
     main()