hysts commited on
Commit
27916d0
1 Parent(s): 8f29722
Files changed (2) hide show
  1. app_training.py +3 -3
  2. 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('Message')
94
- message = gr.Markdown()
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=message)
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('Message')
72
- message = gr.Markdown()
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=message)
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