Update app.py

app.py

@@ -4,6 +4,7 @@ import datetime
 import toml
 import gradio as gr
 from PIL import Image, PngImagePlugin
+from huggingface_hub import HfApi
 
 from pnginfo import read_info_from_image, send_paras
 from images_history import img_history_ui
@@ -12,6 +13,7 @@ from utils import set_token, generate_novelai_image, image_from_bytes
 client_config = toml.load("config.toml")['client']
 today_count = 0
 today = datetime.date.today().strftime('%Y-%m-%d')
+api = HfApi()
 
 def get_count():
     global today_count, today
@@ -118,6 +120,7 @@ def generate(prompt, quality_tags, neg_prompt, seed, scale, width, height, steps
         for k, v in img.info.items():
             pnginfo_data.add_text(k, str(v))
         img.save(os.path.join(today_path, filename), pnginfo=pnginfo_data)
+        api.upload_folder(folder_path=today_path, repo_id="P01yH3dr0n/naiamges", repo_type="dataset")
     return img, payload
 
 def preview_ui():