Update app.py

app.py

@@ -4,7 +4,6 @@ import pandas as pd
 from datetime import datetime
 from PIL import Image
 from PIL import ImageFilter
-from transformers import pipeline
 from io import BytesIO
 import zipfile
 import base64
@@ -74,12 +73,25 @@ def download_photos_as_zip(file_paths):
             unsafe_allow_html=True
         )
 
+# モデルとラベルをダウンロードする関数
+def load_model():
+    path = huggingface_hub.hf_hub_download('public-data/DeepDanbooru', 'model-resnet_custom_v3.h5')
+    model = tf.keras.models.load_model(path)
+    return model
+
+def load_labels():
+    path = huggingface_hub.hf_hub_download('public-data/DeepDanbooru', 'tags.txt')
+    with open(path) as f:
+        labels = [line.strip() for line in f.readlines()]
+    return labels
+
 # Streamlit アプリケーションのメイン部分
 def main():
     st.sidebar.title("アップロードオプション")
     uploaded_photos = st.sidebar.file_uploader("写真をアップロードしてください", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
 
     # モデルとラベルをダウンロードする
+    global model, labels
     model = load_model()
     labels = load_labels()
 
@@ -139,7 +151,4 @@ def main():
         download_photos_as_zip(file_paths)
 
 if __name__ == "__main__":
-    # モデルとラベルをダウンロードする
-    model = load_model()
-    labels = load_labels()
     main()