Update app.py

app.py

@@ -41,7 +41,8 @@ def load_sfts(repo_1_id, repo_2_id):
     if instance_prompt_1 is not None:
         print(f"Trigger word 1: {instance_prompt_1}")
     else:
-        print(f"Trigger word 1: no trigger word need")
+        instance_prompt_1 = "no trigger word needed"
+        print(f"Trigger word 1: no trigger word needed")
 
     card_2 = ModelCard.load(repo_1_id)
     
@@ -50,7 +51,8 @@ def load_sfts(repo_1_id, repo_2_id):
     if instance_prompt_2 is not None:
         print(f"Trigger word 2: {instance_prompt_2}")
     else:
-        print(f"Trigger word 2: no trigger word need")
+        instance_prompt_2 = "no trigger word needed"
+        print(f"Trigger word 2: no trigger word needed")
 
       
     # List all ".safetensors" files in repos
@@ -70,7 +72,7 @@ def load_sfts(repo_1_id, repo_2_id):
     if sfts_available_files_2 == []:
         sfts_available_files_2 = ["NO SAFETENSORS FILE"]
     
-    return gr.update(choices=sfts_available_files_1, value=sfts_available_files_1[0], visible=True), gr.update(choices=sfts_available_files_2, value=sfts_available_files_2[0], visible=True)
+    return gr.update(choices=sfts_available_files_1, value=sfts_available_files_1[0], visible=True), gr.update(choices=sfts_available_files_2, value=sfts_available_files_2[0], visible=True), gr.update(value=instance_prompt_1, visible=True), gr.update(value=instance_prompt_2, visible=True)
     
 @spaces.GPU
 def infer(lora_1_id, lora_1_sfts, lora_2_id, lora_2_sfts, prompt, negative_prompt, lora_1_scale, lora_2_scale, seed):
@@ -230,6 +232,12 @@ with gr.Blocks(css=css) as demo:
                         label = "Safetensors file",
                         visible=False
                     )
+
+                    instance_prompt_1 = gr.Textbox(
+                        label = "Trigger Word 1",
+                        visible = False,
+                        interactive = False
+                    )
                 
                 with gr.Column():
                     
@@ -244,6 +252,12 @@ with gr.Blocks(css=css) as demo:
                         label = "Safetensors file",
                         visible=False
                     )
+
+                    instance_prompt_1 = gr.Textbox(
+                        label = "Trigger Word 1",
+                        visible = False,
+                        interactive = False
+                    )
         else:
             with gr.Row():
                 
@@ -258,6 +272,12 @@ with gr.Blocks(css=css) as demo:
                         visible=False
                     )
 
+                    instance_prompt_2 = gr.Textbox(
+                        label = "Trigger Word 2",
+                        visible = False,
+                        interactive = False
+                    )
+
                 with gr.Column():
                     lora_2_id = gr.Textbox(
                             label = "LoRa 2 ID",
@@ -268,6 +288,12 @@ with gr.Blocks(css=css) as demo:
                         label = "Safetensors file",
                         visible=False
                     )
+
+                    instance_prompt_2 = gr.Textbox(
+                        label = "Trigger Word 2",
+                        visible = False,
+                        interactive = False
+                    )
         
         load_models_btn = gr.Button("1. Load models and .safetensors")
 
@@ -335,7 +361,9 @@ with gr.Blocks(css=css) as demo:
         ],
         outputs = [
             lora_1_sfts,
-            lora_2_sfts
+            lora_2_sfts,
+            instance_prompt_1,
+            instance_prompt_2
         ]
     )
     run_btn.click(