Update app.py

app.py

@@ -47,15 +47,23 @@ 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 main(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+def main(content_b_lora, style_b_lora, prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
         
     generator = torch.Generator().manual_seed(seed)
+
+    if content_b_lora is None:
+        content_B_LoRA_path = ''
+    else:
+        content_B_LoRA_path = content_b_lora
+
+    if style_b_lora is None:
+        style_B_LoRA_path = ''
+    else:
+        style_B_LoRA_path = style_b_lora
     
-    content_B_LoRA_path = ''
-    style_B_LoRA_path = 'fffiloni/b_lora_tardi_4'
     content_alpha,style_alpha = 1,1.1
 
     load_b_lora_to_unet(pipeline, content_B_LoRA_path, style_B_LoRA_path, content_alpha, style_alpha)
@@ -91,6 +99,10 @@ with gr.Blocks(css=css) as demo:
         # Text-to-Image Gradio Template
         Currently running on {power_device}.
         """)
+
+        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.Row():
             
@@ -163,7 +175,7 @@ with gr.Blocks(css=css) as demo:
 
     run_button.click(
         fn = main,
-        inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        inputs = [content_b_lora, style_b_lora, prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs = [result]
     )