Update app.py

app.py

@@ -47,6 +47,29 @@ def load_b_lora_to_unet(pipe, content_lora_model_id: str = '', style_lora_model_
         except Exception as e:
             raise type(e)(f'failed to load_b_lora_to_unet, due to: {e}')
 
+def load_b_loras(content_b_lora, style_b_lora):
+    
+    if content_b_lora is not None:
+        # Get instance_prompt a.k.a trigger word
+        content_model_card = ModelCard.load(custom_model)
+        content_model_repo_data = content_model_card.data.to_dict()
+        content_model_instance_prompt = content_model_repo_data.get("instance_prompt")
+    else:
+        content_model_instance_prompt = ''
+
+    if style_b_lora is not None:
+        # Get instance_prompt a.k.a trigger word
+        style_model_card = ModelCard.load(custom_model)
+        style_model_repo_data = style_model_card.data.to_dict()
+        style_model_instance_prompt = style_model_repo_data.get("instance_prompt")
+        style_model_instance_prompt = f"in {style_model_instance_prompt} style"
+    else:
+        style_model_instance_prompt = ''
+
+    prepared_prompt = f"{content_model_instance_prompt} {style_model_instance_prompt}"
+    
+    return prepared_prompt
+
 def main(content_b_lora, style_b_lora, prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
 
     if randomize_seed:
@@ -100,10 +123,13 @@ with gr.Blocks(css=css) as demo:
         Currently running on {power_device}.
         """)
 
-        with gr.Row():
+        with gr.Row():    
             content_b_lora = gr.Textbox(label="B-LoRa for content")
             style_b_lora = gr.Textbox(label="B-LoRa for style")
         
+        with gr.Column():
+            load_b_loras_btn = gr.Button("load models")
+        
         with gr.Row():
             
             prompt = gr.Text(
@@ -173,6 +199,12 @@ with gr.Blocks(css=css) as demo:
                     value=50,
                 )
 
+    load_b_loras_btn.click(
+        fn = load_b_loras,
+        inputs = [content_b_lora, style_b_lora],
+        outputs = [prompt]
+    )
+    
     run_button.click(
         fn = main,
         inputs = [content_b_lora, style_b_lora, prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],