Update app.py

app.py

@@ -22,12 +22,9 @@ pipeline = StableDiffusionXLPipeline.from_pretrained(
 pipeline.enable_model_cpu_offload() 
 pipeline.enable_vae_slicing()
 
-# DDIM inversion
-
-global zts, src_style, src_prompt
-
-def load_refs(ref_path, ref_style, ref_prompt, prompt1, prompt2, prompt3):
-
+def run(ref_path, ref_style, ref_prompt, prompt1, prompt2, prompt3):
+    
+    # DDIM inversion
     src_style = f"{ref_style}"
     src_prompt = f"{ref_prompt}, {src_style}."
     image_path = f"{ref_path}"
@@ -36,9 +33,7 @@ def load_refs(ref_path, ref_style, ref_prompt, prompt1, prompt2, prompt3):
     x0 = np.array(load_image(image_path).resize((1024, 1024)))
     zts = inversion.ddim_inversion(pipeline, x0, src_prompt, num_inference_steps, 2)
     #mediapy.show_image(x0, title="innput reference image", height=256)
-    return "ready"
 
-def run(prompt1, prompt2, prompt3):
     # run StyleAligned
     prompts = [
         src_prompt,
@@ -94,27 +89,16 @@ with gr.Blocks() as demo:
             ref_path = gr.Image(type="filepath", value="./example_image/medieval-bed.jpeg")
             ref_style = gr.Textbox(label="Reference style", value="medieval painting")
             ref_prompt = gr.Textbox(label="Referéce prompt", value="Man laying on bed")
-            load_ref_btn = gr.Button("Load Reference")
+            
         with gr.Column():
-            is_ready = gr.Textbox(label = "Ref status")
+            
             prompt1 = gr.Textbox(label="Prompt1", value="A man working on a laptop")
             prompt2 = gr.Textbox(label="Prompt2", value="A man eating pizza")
             prompt3 = gr.Textbox(label="Prompt3", value="A woman playing on saxophone")
             run_button = gr.Button("Submit")
         with gr.Column():
             results = gr.Gallery()
-    load_ref_btn.click(
-        fn = load_refs,
-        inputs = [
-            ref_path,
-            ref_style,
-            ref_prompt
-        ],
-        outputs = [
-            is_ready
-        ],
-        queue = False
-    )
+    
     run_button.click(
         fn = run,
         inputs = [