hysts HF staff commited on
Commit
8f8e574
1 Parent(s): 46ce1fe
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -115,8 +115,8 @@ with gr.Blocks(css='style.css') as demo:
115
  check_base_model_button = gr.Button('Check current base model')
116
  with gr.Row():
117
  with gr.Column():
118
- base_model_id = gr.Text(
119
- label='Base model repo',
120
  max_lines=1,
121
  placeholder='runwayml/stable-diffusion-v1-5',
122
  info=
@@ -136,11 +136,11 @@ with gr.Blocks(css='style.css') as demo:
136
  check_base_model_button.click(fn=lambda: model.base_model_id,
137
  outputs=current_base_model,
138
  queue=False)
139
- base_model_id.submit(fn=model.set_base_model,
140
- inputs=base_model_id,
141
- outputs=current_base_model)
142
  change_base_model_button.click(fn=model.set_base_model,
143
- inputs=base_model_id,
144
  outputs=current_base_model)
145
 
146
  demo.queue(api_open=False).launch()
 
115
  check_base_model_button = gr.Button('Check current base model')
116
  with gr.Row():
117
  with gr.Column():
118
+ new_base_model_id = gr.Text(
119
+ label='New base model',
120
  max_lines=1,
121
  placeholder='runwayml/stable-diffusion-v1-5',
122
  info=
 
136
  check_base_model_button.click(fn=lambda: model.base_model_id,
137
  outputs=current_base_model,
138
  queue=False)
139
+ new_base_model_id.submit(fn=model.set_base_model,
140
+ inputs=new_base_model_id,
141
+ outputs=current_base_model)
142
  change_base_model_button.click(fn=model.set_base_model,
143
+ inputs=new_base_model_id,
144
  outputs=current_base_model)
145
 
146
  demo.queue(api_open=False).launch()