add lora configs as gradio parameters

README.md

@@ -1,5 +1,5 @@
 ---
-title: FLUX.1 [Inpainting]
+title: FLUX.1 [Inpainting with lora]
 emoji: 🎨
 colorFrom: yellow
 colorTo: pink

app.py

@@ -22,8 +22,8 @@ IMAGE_SIZE = 1024
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 trigger_word = "a_photo_of_TOK"
-lora_path = "jiuface/boy-001"
-lora_weights = "flux_train_replicate.safetensors"
+# lora_path = "jiuface/boy-001"
+# lora_weights = "flux_train_replicate.safetensors"
 lora_scale = 0.9
 
 login(token=HF_TOKEN)
@@ -97,6 +97,8 @@ def resize_image_dimensions(
 @spaces.GPU(duration=100)
 def process(
     input_image_editor: dict,
+    lora_path: str,
+    lora_weights: str,
     input_text: str,
     seed_slicer: int,
     randomize_seed_checkbox: bool,
@@ -123,6 +125,7 @@ def process(
         width, height = resize_image_dimensions(original_resolution_wh=image.size)
         resized_image = image.resize((width, height), Image.LANCZOS)
         resized_mask = mask.resize((width, height), Image.LANCZOS)
+    
     with calculateDuration("load lora"):
         pipe.load_lora_weights(lora_path, weight_name=lora_weights)
     
@@ -157,15 +160,32 @@ with gr.Blocks() as demo:
                 image_mode='RGB',
                 layers=False,
                 brush=gr.Brush(colors=["#FFFFFF"], color_mode="fixed"))
+            
+            with gr.Accordion("Lora Settings", open=True):
+                lora_path = gr.Text(
+                    label="Lora path",
+                    show_label=True,
+                    max_lines=1,
+                    placeholder="Enter your lora path",
+                    container=False,
+                )
+                lora_weights = gr.Text(
+                    label="Lora weights",
+                    show_label=True,
+                    max_lines=1,
+                    placeholder="Enter your lora weights",
+                    container=False,
+                )
 
             with gr.Row():
                 input_text_component = gr.Text(
                     label="Prompt",
-                    show_label=False,
+                    show_label=True,
                     max_lines=1,
                     placeholder="Enter your prompt",
                     container=False,
                 )
+                
                 submit_button_component = gr.Button(
                     value='Submit', variant='primary', scale=0)
 
@@ -213,6 +233,8 @@ with gr.Blocks() as demo:
         fn=process,
         inputs=[
             input_image_editor_component,
+            lora_path,
+            lora_weights,
             input_text_component,
             seed_slicer_component,
             randomize_seed_checkbox_component,