fffiloni commited on
Commit
02cde20
·
1 Parent(s): 54c5b48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +54 -37
app.py CHANGED
@@ -57,8 +57,10 @@ def load_model(custom_model):
57
 
58
  print(f"Safetensors available: {sfts_available_files}")
59
 
60
- return gr.update(choices=sfts_available_files, value=sfts_available_files[0], visible=True), gr.update(value=instance_prompt, visible=True)
61
 
 
 
62
 
63
  @spaces.GPU
64
  def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):
@@ -187,52 +189,67 @@ with gr.Blocks(css=css) as demo:
187
  )
188
  with gr.Column():
189
  load_model_btn = gr.Button("Load my model")
 
 
 
 
190
  trigger_word = gr.Textbox(label="Trigger word", interactive=False, visible=False)
191
 
192
- prompt_in = gr.Textbox(label="Prompt")
193
- with gr.Row():
194
- inf_steps = gr.Slider(
195
- label="Inference steps",
196
- minimum=12,
197
- maximum=50,
198
- step=1,
199
- value=25
200
- )
201
- guidance_scale = gr.Slider(
202
- label="Guidance scale",
203
- minimum=0.0,
204
- maximum=50.0,
205
- step=0.1,
206
- value=7.5
207
- )
208
 
209
- with gr.Row():
210
- seed = gr.Slider(
211
- label="Seed",
212
- info = "-1 denotes a random seed",
213
- minimum=-1,
214
- maximum=423538377342,
215
- step=1,
216
- value=-1
217
- )
218
- last_used_seed = gr.Number(
219
- label = "Last used seed",
220
- info = "the seed used in the last generation",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
221
  )
222
- lora_weight = gr.Slider(
223
- label="LoRa weigth",
224
- minimum=0.0,
225
- maximum=1.0,
226
- step=0.01,
227
- value=0.9
228
- )
229
  submit_btn = gr.Button("Submit")
230
  image_out = gr.Image(label="Image output")
231
 
 
 
 
 
 
 
232
  load_model_btn.click(
233
  fn = load_model,
234
  inputs=[custom_model],
235
- outputs = [weight_name, trigger_word],
236
  queue = False
237
  )
238
  submit_btn.click(
 
57
 
58
  print(f"Safetensors available: {sfts_available_files}")
59
 
60
+ return "Model Ready", gr.update(choices=sfts_available_files, value=sfts_available_files[0], visible=True), gr.update(value=instance_prompt, visible=True)
61
 
62
+ def custom_model_changed(custom_model):
63
+ return "Model changed, you must reload before re-run"
64
 
65
  @spaces.GPU
66
  def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):
 
189
  )
190
  with gr.Column():
191
  load_model_btn = gr.Button("Load my model")
192
+ model_status = gr.Textbox(
193
+ label = "model status",
194
+ show_label = False
195
+ )
196
  trigger_word = gr.Textbox(label="Trigger word", interactive=False, visible=False)
197
 
198
+ prompt_in = gr.Textbox(
199
+ label="Your Prompt",
200
+ info = "Dont' forget to include your trigger word if necessary"
201
+ )
 
 
 
 
 
 
 
 
 
 
 
 
202
 
203
+ with gr.Accordion("Advanced Settings", open=False):
204
+ with gr.Row():
205
+ inf_steps = gr.Slider(
206
+ label="Inference steps",
207
+ minimum=12,
208
+ maximum=50,
209
+ step=1,
210
+ value=25
211
+ )
212
+ guidance_scale = gr.Slider(
213
+ label="Guidance scale",
214
+ minimum=0.0,
215
+ maximum=50.0,
216
+ step=0.1,
217
+ value=7.5
218
+ )
219
+
220
+ with gr.Row():
221
+ seed = gr.Slider(
222
+ label="Seed",
223
+ info = "-1 denotes a random seed",
224
+ minimum=-1,
225
+ maximum=423538377342,
226
+ step=1,
227
+ value=-1
228
+ )
229
+ last_used_seed = gr.Number(
230
+ label = "Last used seed",
231
+ info = "the seed used in the last generation",
232
+ )
233
+ lora_weight = gr.Slider(
234
+ label="LoRa weigth",
235
+ minimum=0.0,
236
+ maximum=1.0,
237
+ step=0.01,
238
+ value=0.9
239
  )
 
 
 
 
 
 
 
240
  submit_btn = gr.Button("Submit")
241
  image_out = gr.Image(label="Image output")
242
 
243
+ custom_model.blur(
244
+ fn=custom_model_changed,
245
+ inputs = [custom_model],
246
+ outputs = [model_status],
247
+ queue = False
248
+ )
249
  load_model_btn.click(
250
  fn = load_model,
251
  inputs=[custom_model],
252
+ outputs = [model_status, weight_name, trigger_word],
253
  queue = False
254
  )
255
  submit_btn.click(