Update app.py

app.py

@@ -4,7 +4,7 @@ import datetime
 import toml
 import gradio as gr
 from PIL import Image, PngImagePlugin
-from huggingface_hub import HfApi, hf_hub_download
+from huggingface_hub import HfApi, snapshot_download
 
 from pnginfo import read_info_from_image, send_paras
 from images_history import img_history_ui
@@ -196,7 +196,7 @@ def ui():
 
 
 if __name__ == '__main__':
-    hf_hub_download(repo_id="P01yH3dr0n/naimages", subfolder="data_client", repo_type="dataset", local_dir="./", token=os.environ.get("hf_token"))
+    snapshot_download(repo_id="P01yH3dr0n/naimages", repo_type="dataset", local_dir="./", token=os.environ.get("hf_token"), allow_patterns="*.png")
     website = ui()
     website.queue(default_concurrency_limit=5)
     website.launch(auth=(os.environ.get('account'), os.environ.get('password')), allowed_paths=['tagcomplete'], debug=True)