Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -162,7 +162,7 @@ def generate_image(input_image, prompt, selected_index, user_lora, seed=42, rand
|
|
162 |
if lora_to_use and lora_to_use != active_lora_adapter:
|
163 |
try:
|
164 |
if active_lora_adapter:
|
165 |
-
pipe
|
166 |
|
167 |
lora_path = load_lora_weights(lora_to_use["repo"], lora_to_use["weights"])
|
168 |
if lora_path:
|
@@ -241,12 +241,11 @@ css = """
|
|
241 |
"""
|
242 |
|
243 |
# Build the Gradio interface
|
244 |
-
with gr.Blocks(css=css) as demo:
|
245 |
gr_lora_configs = gr.State(value=lora_configs)
|
246 |
|
247 |
title = gr.HTML(
|
248 |
-
"""<h1>
|
249 |
-
<p>Edit images using custom style adapters. Fast generation with minimal steps.</p>""",
|
250 |
)
|
251 |
|
252 |
selected_state = gr.State(value=None)
|
|
|
162 |
if lora_to_use and lora_to_use != active_lora_adapter:
|
163 |
try:
|
164 |
if active_lora_adapter:
|
165 |
+
pipe.unload_lora_weights()
|
166 |
|
167 |
lora_path = load_lora_weights(lora_to_use["repo"], lora_to_use["weights"])
|
168 |
if lora_path:
|
|
|
241 |
"""
|
242 |
|
243 |
# Build the Gradio interface
|
244 |
+
with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 60)) as demo:
|
245 |
gr_lora_configs = gr.State(value=lora_configs)
|
246 |
|
247 |
title = gr.HTML(
|
248 |
+
"""<h1>Flux Kontext DLC😍</h1>""",
|
|
|
249 |
)
|
250 |
|
251 |
selected_state = gr.State(value=None)
|