Upload 2 files

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import os
-from convert_repo_to_safetensors_gr import convert_repo_to_safetensors_multi
+from convert_repo_to_safetensors_gr import convert_repo_to_safetensors_multi, clear_safetensors
 os.environ['HF_OUTPUT_REPO'] = 'John6666/safetensors_converting_test'
 
 css = """"""
@@ -24,6 +24,7 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css, delete_ca
         run_button = gr.Button(value="Convert")
         st_file = gr.Files(label="Output", interactive=False)
         st_md = gr.Markdown()
+        delete_button = gr.Button(value="Delete Safetensors")
 
     gr.on(
         triggers=[repo_id.submit, run_button.click],
@@ -31,6 +32,7 @@ with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css, delete_ca
         inputs=[repo_id, hf_token, st_file, uploaded_urls, dtype, is_upload, newrepo_id, newrepo_type, is_private],
         outputs=[st_file, uploaded_urls, st_md],
     )
+    delete_button.click(clear_safetensors, None, [st_file], queue=False, show_api=False)
 
 demo.queue()
 demo.launch()

convert_repo_to_safetensors_gr.py

@@ -13,6 +13,8 @@ import gradio as gr
 from huggingface_hub import HfApi, HfFolder, hf_hub_url, snapshot_download
 import os
 from pathlib import Path
+import shutil
+import gc
 
 def get_token():
     try:
@@ -388,6 +390,7 @@ def convert_repo_to_safetensors(repo_id, dtype="fp16", progress=gr.Progress(trac
     progress(0, desc="Start converting...")
     convert_diffusers_to_safetensors(download_dir, output_filename, dtype)
     progress(1, desc="Converted.")
+    shutil.rmtree(download_dir)
     return output_filename
 
 
@@ -408,9 +411,18 @@ def convert_repo_to_safetensors_multi(repo_id, hf_token, files, urls, dtype="fp1
         md += f"[Download {str(u).split('/')[-1]}]({str(u)})<br>"
     if not files: files = []
     files.append(file)
+    gc.collect()
     return gr.update(value=files), gr.update(value=urls, choices=urls), gr.update(value=md)
 
 
+def clear_safetensors():
+    for p in Path('.').glob('*.safetensors'):
+        p.unlink()
+        print("Deleted.")
+    gc.collect()
+    return gr.update(value=[])
+
+
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()