hysts
commited on
Commit
·
27916d0
1
Parent(s):
8f29722
Rename
Browse files- app_training.py +3 -3
- app_upload.py +3 -3
app_training.py
CHANGED
|
@@ -90,8 +90,8 @@ def create_training_demo(trainer: Trainer,
|
|
| 90 |
run_button = gr.Button('Start Training')
|
| 91 |
|
| 92 |
with gr.Box():
|
| 93 |
-
gr.Markdown('
|
| 94 |
-
|
| 95 |
|
| 96 |
if pipe is not None:
|
| 97 |
run_button.click(fn=pipe.clear)
|
|
@@ -118,7 +118,7 @@ def create_training_demo(trainer: Trainer,
|
|
| 118 |
delete_existing_repo,
|
| 119 |
upload_to,
|
| 120 |
],
|
| 121 |
-
outputs=
|
| 122 |
return demo
|
| 123 |
|
| 124 |
|
|
|
|
| 90 |
run_button = gr.Button('Start Training')
|
| 91 |
|
| 92 |
with gr.Box():
|
| 93 |
+
gr.Markdown('Output message')
|
| 94 |
+
output_message = gr.Markdown()
|
| 95 |
|
| 96 |
if pipe is not None:
|
| 97 |
run_button.click(fn=pipe.clear)
|
|
|
|
| 118 |
delete_existing_repo,
|
| 119 |
upload_to,
|
| 120 |
],
|
| 121 |
+
outputs=output_message)
|
| 122 |
return demo
|
| 123 |
|
| 124 |
|
app_upload.py
CHANGED
|
@@ -68,8 +68,8 @@ def create_upload_demo(hf_token: str | None) -> gr.Blocks:
|
|
| 68 |
- You can upload your trained model to your personal profile (i.e. https://huggingface.co/{your_username}/{model_name}) or to the public [LoRA Concepts Library](https://huggingface.co/lora-library) (i.e. https://huggingface.co/lora-library/{model_name}).
|
| 69 |
''')
|
| 70 |
with gr.Box():
|
| 71 |
-
gr.Markdown('
|
| 72 |
-
|
| 73 |
|
| 74 |
reload_button.click(fn=load_local_lora_model_list,
|
| 75 |
inputs=None,
|
|
@@ -82,7 +82,7 @@ def create_upload_demo(hf_token: str | None) -> gr.Blocks:
|
|
| 82 |
use_private_repo,
|
| 83 |
delete_existing_repo,
|
| 84 |
],
|
| 85 |
-
outputs=
|
| 86 |
|
| 87 |
return demo
|
| 88 |
|
|
|
|
| 68 |
- You can upload your trained model to your personal profile (i.e. https://huggingface.co/{your_username}/{model_name}) or to the public [LoRA Concepts Library](https://huggingface.co/lora-library) (i.e. https://huggingface.co/lora-library/{model_name}).
|
| 69 |
''')
|
| 70 |
with gr.Box():
|
| 71 |
+
gr.Markdown('Output message')
|
| 72 |
+
output_message = gr.Markdown()
|
| 73 |
|
| 74 |
reload_button.click(fn=load_local_lora_model_list,
|
| 75 |
inputs=None,
|
|
|
|
| 82 |
use_private_repo,
|
| 83 |
delete_existing_repo,
|
| 84 |
],
|
| 85 |
+
outputs=output_message)
|
| 86 |
|
| 87 |
return demo
|
| 88 |
|