Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -196,7 +196,7 @@ def delete_ckpt():
|
|
| 196 |
os.system(f"rm -rf ckpt")
|
| 197 |
return "delete ckpt done!"
|
| 198 |
|
| 199 |
-
def
|
| 200 |
# Create the ckpt directory if it doesn't exist
|
| 201 |
os.makedirs("ckpt", exist_ok=True)
|
| 202 |
|
|
@@ -480,11 +480,11 @@ with block:
|
|
| 480 |
text_safetensors_token = gr.Textbox(show_label=False, max_lines=1, placeholder="🤗 token")
|
| 481 |
out_safetensors = gr.Textbox(show_label=False)
|
| 482 |
with gr.Row().style(equal_height=True):
|
| 483 |
-
|
| 484 |
btn_to_safetensors = gr.Button("Convert to Safetensors")
|
| 485 |
btn_push_safetensors = gr.Button("Push Safetensors to 🤗")
|
| 486 |
btn_delete_safetensors = gr.Button("Delete Safetensors")
|
| 487 |
-
|
| 488 |
btn_to_safetensors.click(to_safetensors, inputs=[text_ckpt_name_2, text_safetensors_name_2], outputs=out_safetensors)
|
| 489 |
btn_push_safetensors.click(push_safetensors, inputs=[text_safetensors_model_to, text_safetensors_token, text_safetensors_branch], outputs=out_safetensors)
|
| 490 |
btn_delete_safetensors.click(delete_safetensors, outputs=out_safetensors)
|
|
|
|
| 196 |
os.system(f"rm -rf ckpt")
|
| 197 |
return "delete ckpt done!"
|
| 198 |
|
| 199 |
+
def download_ckpt_1(ckpt_url_2, ckpt_name_2):
|
| 200 |
# Create the ckpt directory if it doesn't exist
|
| 201 |
os.makedirs("ckpt", exist_ok=True)
|
| 202 |
|
|
|
|
| 480 |
text_safetensors_token = gr.Textbox(show_label=False, max_lines=1, placeholder="🤗 token")
|
| 481 |
out_safetensors = gr.Textbox(show_label=False)
|
| 482 |
with gr.Row().style(equal_height=True):
|
| 483 |
+
btn_download_ckpt_1 = gr.Button("Download CKPT")
|
| 484 |
btn_to_safetensors = gr.Button("Convert to Safetensors")
|
| 485 |
btn_push_safetensors = gr.Button("Push Safetensors to 🤗")
|
| 486 |
btn_delete_safetensors = gr.Button("Delete Safetensors")
|
| 487 |
+
btn_download_ckpt_1.click(download_ckpt_1, inputs=[text_ckpt_url_2, text_ckpt_name_2], outputs=out_safetensors)
|
| 488 |
btn_to_safetensors.click(to_safetensors, inputs=[text_ckpt_name_2, text_safetensors_name_2], outputs=out_safetensors)
|
| 489 |
btn_push_safetensors.click(push_safetensors, inputs=[text_safetensors_model_to, text_safetensors_token, text_safetensors_branch], outputs=out_safetensors)
|
| 490 |
btn_delete_safetensors.click(delete_safetensors, outputs=out_safetensors)
|