Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -123,6 +123,13 @@ def create_interface():
|
|
123 |
"Snoopy", "Van_Gogh", "Vector"
|
124 |
]
|
125 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
126 |
# Example data
|
127 |
examples = [
|
128 |
["3D_Chibi", "", "3D Chibi style, Two smiling colleagues enthusiastically high-five in front of a whiteboard filled with technical notes about multimodal learning, reflecting a moment of success and collaboration at OpenAI.",
|
@@ -154,7 +161,7 @@ def create_interface():
|
|
154 |
with gr.Column(scale=1):
|
155 |
image_input = gr.Image(type="pil", label="Upload Image")
|
156 |
prompt_box = gr.Textbox(label="Prompt",
|
157 |
-
|
158 |
info="Remember to include the necessary trigger words if you're using a custom LoRA."
|
159 |
)
|
160 |
lora_dropdown = gr.Dropdown(
|
@@ -177,6 +184,9 @@ def create_interface():
|
|
177 |
seed_slider = gr.Slider(
|
178 |
1, 2_147_483_647, value=42, step=1, label="Seed")
|
179 |
|
|
|
|
|
|
|
180 |
gr.Examples(
|
181 |
examples=examples,
|
182 |
inputs=[lora_dropdown, custom_repo_box, prompt_box, image_input,
|
|
|
123 |
"Snoopy", "Van_Gogh", "Vector"
|
124 |
]
|
125 |
|
126 |
+
def update_trigger_word(lora_name, prompt):
|
127 |
+
for name in demo_lora_names:
|
128 |
+
trigger = " ".join(name.split("_")) + " style,"
|
129 |
+
prompt = prompt.replace(trigger, "")
|
130 |
+
new_trigger = " ".join(lora_name.split("_"))+ " style,"
|
131 |
+
return new_trigger + prompt
|
132 |
+
|
133 |
# Example data
|
134 |
examples = [
|
135 |
["3D_Chibi", "", "3D Chibi style, Two smiling colleagues enthusiastically high-five in front of a whiteboard filled with technical notes about multimodal learning, reflecting a moment of success and collaboration at OpenAI.",
|
|
|
161 |
with gr.Column(scale=1):
|
162 |
image_input = gr.Image(type="pil", label="Upload Image")
|
163 |
prompt_box = gr.Textbox(label="Prompt",
|
164 |
+
value="3D Chibi style,",
|
165 |
info="Remember to include the necessary trigger words if you're using a custom LoRA."
|
166 |
)
|
167 |
lora_dropdown = gr.Dropdown(
|
|
|
184 |
seed_slider = gr.Slider(
|
185 |
1, 2_147_483_647, value=42, step=1, label="Seed")
|
186 |
|
187 |
+
lora_dropdown.select(fn=update_trigger_word, inputs=[lora_dropdown,prompt_box],
|
188 |
+
outputs=prompt_box)
|
189 |
+
|
190 |
gr.Examples(
|
191 |
examples=examples,
|
192 |
inputs=[lora_dropdown, custom_repo_box, prompt_box, image_input,
|