miaw1419 commited on
Commit
726c69b
·
verified ·
1 Parent(s): 4f43190

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -472,12 +472,16 @@ def api_fn(api):
472
  gr.Warning("Invalid API!")
473
  valid_api = ""
474
 
 
 
 
 
475
 
 
476
 
477
  @spaces.GPU(duration=45)
478
  def generate(prompt, vp_pos, vp_neg, slider, example_prompt, gallery, num_inference_steps, seed, randomize_seed):
479
  if vp_pos == "" and vp_neg == "":
480
- gr.Warning("Visual preference is empty.", duration=10)
481
  slider = 0
482
 
483
  for filter in word_list:
@@ -754,7 +758,7 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
754
  )
755
 
756
  run_button.click(
757
- fn = generate,
758
  inputs = [prompt, positive_extracted_vp, negative_extracted_vp, slider, example_prompt, gallery, num_inference_steps, seed, randomize_seed],
759
  outputs = [result, example_prompt, gallery],
760
  )
 
472
  gr.Warning("Invalid API!")
473
  valid_api = ""
474
 
475
+
476
+ def generate_out(prompt, vp_pos, vp_neg, slider, example_prompt, gallery, num_inference_steps, seed, randomize_seed):
477
+ if vp_pos == "" and vp_neg == "":
478
+ gr.Warning("Visual preference is empty.", duration=10)
479
 
480
+ return generate(prompt, vp_pos, vp_neg, slider, example_prompt, gallery, num_inference_steps, seed, randomize_seed)
481
 
482
  @spaces.GPU(duration=45)
483
  def generate(prompt, vp_pos, vp_neg, slider, example_prompt, gallery, num_inference_steps, seed, randomize_seed):
484
  if vp_pos == "" and vp_neg == "":
 
485
  slider = 0
486
 
487
  for filter in word_list:
 
758
  )
759
 
760
  run_button.click(
761
+ fn = generate_out,
762
  inputs = [prompt, positive_extracted_vp, negative_extracted_vp, slider, example_prompt, gallery, num_inference_steps, seed, randomize_seed],
763
  outputs = [result, example_prompt, gallery],
764
  )