Script ini secara otomatis memantau direktori output untuk file gambar baru dan mengunggahnya ke repository Hugging Face. Script ini dirancang untuk bekerja dengan gambar yang dihasilkan oleh Stable Diffusion.
- Pemantauan Direktori: Memantau direktori tertentu secara otomatis untuk file baru yang dibuat.
- Penanganan File: Mendeteksi file
.tmpsementara, menunggu hingga file tersebut dikonversi menjadi.png, lalu mengunggah file.pngke Hugging Face. - Integrasi dengan Hugging Face: Mengunggah file langsung ke repository Hugging Face, mendukung tipe repository model dan dataset.
- Antarmuka Gradio: Menyediakan antarmuka web yang intuitif untuk memulai pemantauan dan pengunggahan file.
Di antarmuka Gradio:
- Masukkan token Hugging Face Anda.
- Pilih tipe repository (
modelataudataset). - Masukkan ID repository tempat file akan diunggah.
- Klik tombol "Start Monitoring".
- Pemantauan File: Script ini menggunakan library Watchdog untuk memantau direktori tertentu (
/kaggle/working/stable-diffusion-webui/outputs/txt2img-images/2024-08-07) untuk file baru. - Pengunggahan File: Saat file baru terdeteksi, script akan memeriksa formatnya:
- Jika file berformat
.tmp, script akan menunggu beberapa detik dan memeriksa apakah file tersebut berubah menjadi.png. Jika ya, file.pngakan diunggah ke repository Hugging Face. - Jika file sudah dalam format
.png, file tersebut akan langsung diunggah.
- Jika file berformat
- Direktori yang dipantau dapat diubah dengan memodifikasi
pathdi fungsistart_watcher. - Script ini mendukung berbagai tipe repository (
model,dataset). Pastikan tipe repository yang benar dipilih di antarmuka Gradio@.
@jekverse