seawolf2357 commited on
Commit
58a26da
·
verified ·
1 Parent(s): 09e5977

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -6
app.py CHANGED
@@ -293,12 +293,16 @@ def update_pricing(steps, oauth_token: Union[gr.OAuthToken, None]):
293
  return gr.update(visible=True), cost_preview, gr.update(visible=False), gr.update(visible=True)
294
  else:
295
  pay_disclaimer = f'''## ⚠️ {user["name"]}, your account doesn't have a payment method. Set one up <a href='https://huggingface.co/settings/billing/payment' target='_blank'>here</a> and come back here to train your LoRA<br><br>'''
296
- return gr.update(visible=True), pay_disclaimer+cost_preview, gr.update(visible=True), gr.update(visible=False)
 
 
 
 
297
  else:
298
  return gr.update(visible=False), "", gr.update(visible=False), gr.update(visible=True)
299
 
300
  def swap_base_model(model):
301
- return gr.update(visible=True) if model == "[dev] (high quality model, non-commercial license)" else gr.update(visible=False)
302
 
303
  config_yaml = '''
304
  device: cuda:0
@@ -374,9 +378,14 @@ with gr.Blocks(theme=theme, css=css) as demo:
374
  placeholder="uncommon word like p3rs0n or trtcrd, or sentence like 'in the style of CNSTLL'",
375
  interactive=True,
376
  )
377
- which_model = gr.Radio(["[schnell] (4 step fast model)", "[dev] (high quality model, non-commercial license - available when training locally)"], label="Which base model to train?", elem_id="space_model" if is_spaces else "local_model", value="[schnell] (4 step fast model)",)
 
 
 
 
 
378
  model_warning = gr.Markdown("""> [dev] model license is non-commercial. By choosing to fine-tune [dev], you must agree with [its license](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md) and make sure the LoRA you will train and the training process you would start does not violate it.
379
- """, visible=False)
380
  with gr.Group(visible=True) as image_upload:
381
  with gr.Row():
382
  images = gr.File(
@@ -468,7 +477,6 @@ with gr.Blocks(theme=theme, css=css) as demo:
468
  pip3 install -r requirements.txt
469
  cd ..
470
  ```
471
-
472
  Login with Hugging Face to access FLUX.1 [dev], choose a token with `write` permissions to push your LoRAs to the HF Hub
473
  ```bash
474
  huggingface-cli login
@@ -545,4 +553,4 @@ with gr.Blocks(theme=theme, css=css) as demo:
545
  demo.load(fn=swap_visibilty, outputs=main_ui)
546
 
547
  if __name__ == "__main__":
548
- demo.launch(share=True, show_error=True)
 
293
  return gr.update(visible=True), cost_preview, gr.update(visible=False), gr.update(visible=True)
294
  else:
295
  pay_disclaimer = f'''## ⚠️ {user["name"]}, your account doesn't have a payment method. Set one up <a href='https://huggingface.co/settings/billing/payment' target='_blank'>here</a> and come back here to train your LoRA<br><br>'''
296
+ return gr.update(visible=True),
297
+
298
+
299
+
300
+ return gr.update(visible=True), pay_disclaimer+cost_preview, gr.update(visible=True), gr.update(visible=False)
301
  else:
302
  return gr.update(visible=False), "", gr.update(visible=False), gr.update(visible=True)
303
 
304
  def swap_base_model(model):
305
+ return gr.update(visible=False) if model == "[schnell] (4 step fast model)" else gr.update(visible=True)
306
 
307
  config_yaml = '''
308
  device: cuda:0
 
378
  placeholder="uncommon word like p3rs0n or trtcrd, or sentence like 'in the style of CNSTLL'",
379
  interactive=True,
380
  )
381
+ which_model = gr.Radio(
382
+ ["[dev] (high quality model, non-commercial license)", "[schnell] (4 step fast model)"],
383
+ label="Which base model to train?",
384
+ elem_id="space_model" if is_spaces else "local_model",
385
+ value="[dev] (high quality model, non-commercial license)",
386
+ )
387
  model_warning = gr.Markdown("""> [dev] model license is non-commercial. By choosing to fine-tune [dev], you must agree with [its license](https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md) and make sure the LoRA you will train and the training process you would start does not violate it.
388
+ """, visible=True)
389
  with gr.Group(visible=True) as image_upload:
390
  with gr.Row():
391
  images = gr.File(
 
477
  pip3 install -r requirements.txt
478
  cd ..
479
  ```
 
480
  Login with Hugging Face to access FLUX.1 [dev], choose a token with `write` permissions to push your LoRAs to the HF Hub
481
  ```bash
482
  huggingface-cli login
 
553
  demo.load(fn=swap_visibilty, outputs=main_ui)
554
 
555
  if __name__ == "__main__":
556
+ demo.launch(share=True, show_error=True)