alexnasa commited on
Commit
2dfc048
·
verified ·
1 Parent(s): 9b99398

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -371,7 +371,7 @@ def replace_attention_processor(unet, clear=False, blur_sigma=None):
371
  @spaces.GPU()
372
  def apply_prompt(meta_data, new_prompt):
373
 
374
- pipe, _, scheduler, _, _ = load_pipeline()
375
  pipe.to("cuda")
376
 
377
  caption, real_latents_cpu, inversed_latents_cpu, saved_weights = meta_data
@@ -646,7 +646,7 @@ with gr.Blocks(analytics_enabled=False) as demo:
646
  lambda: gr.update(interactive=True), outputs=new_prompt_input
647
  )
648
 
649
- image_input.change(fn=choose_caption, inputs=image_input, outputs=[prompt_input])
650
 
651
  steps_slider.release(update_step, inputs=steps_slider)
652
  interpolate_slider.release(update_scale, inputs=interpolate_slider)
 
371
  @spaces.GPU()
372
  def apply_prompt(meta_data, new_prompt):
373
 
374
+ pipe, _, scheduler = load_pipeline()
375
  pipe.to("cuda")
376
 
377
  caption, real_latents_cpu, inversed_latents_cpu, saved_weights = meta_data
 
646
  lambda: gr.update(interactive=True), outputs=new_prompt_input
647
  )
648
 
649
+ image_input.upload(fn=choose_caption, inputs=image_input, outputs=[prompt_input])
650
 
651
  steps_slider.release(update_step, inputs=steps_slider)
652
  interpolate_slider.release(update_scale, inputs=interpolate_slider)