Spaces:
Sleeping
Sleeping
rafaaa2105
commited on
Commit
•
e47b9ec
1
Parent(s):
a7cc301
Add a refresh button to the dropdowns
Browse files
app.py
CHANGED
@@ -74,6 +74,9 @@ def download_civitai_model(model_id, lora_id="", progress=gr.Progress(track_tqdm
|
|
74 |
except Exception as e:
|
75 |
return f"Error downloading model or LoRA: {e}"
|
76 |
|
|
|
|
|
|
|
77 |
def load_model(model, lora="", use_lora=False):
|
78 |
try:
|
79 |
print(f"\n\nLoading {model}...")
|
@@ -145,6 +148,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
|
|
145 |
with gr.Group(elem_id="input_group"):
|
146 |
model_dropdown = gr.Dropdown(choices=models_list, value=models_list[0] if models_list else None, label="Model", elem_id="model_dropdown")
|
147 |
lora_dropdown = gr.Dropdown(choices=loras_list, value=loras_list[0], label="LoRA")
|
|
|
148 |
prompt = gr.Textbox(label="Prompt", elem_id="prompt_textbox")
|
149 |
generate_btn = gr.Button("Generate Image", elem_id="generate_button")
|
150 |
with gr.Accordion("Advanced", open=False, elem_id="advanced_accordion"):
|
@@ -157,6 +161,7 @@ with gr.Blocks(theme='ParityError/Interstellar') as demo:
|
|
157 |
with gr.Column(elem_id="output_column"):
|
158 |
output_gallery = gr.Gallery(label="Generated Images", height=480, scale=1, elem_id="output_gallery")
|
159 |
|
|
|
160 |
generate_btn.click(generate_images, inputs=[model_dropdown, lora_dropdown, prompt, negative_prompt, num_inference_steps, guidance_scale, height, width, num_images], outputs=output_gallery)
|
161 |
|
162 |
with gr.Tab("Download Custom Model"):
|
|
|
74 |
except Exception as e:
|
75 |
return f"Error downloading model or LoRA: {e}"
|
76 |
|
77 |
+
def refresh_dropdowns():
|
78 |
+
return gr.update(choices=models_list), gr.update(choices=loras_list)
|
79 |
+
|
80 |
def load_model(model, lora="", use_lora=False):
|
81 |
try:
|
82 |
print(f"\n\nLoading {model}...")
|
|
|
148 |
with gr.Group(elem_id="input_group"):
|
149 |
model_dropdown = gr.Dropdown(choices=models_list, value=models_list[0] if models_list else None, label="Model", elem_id="model_dropdown")
|
150 |
lora_dropdown = gr.Dropdown(choices=loras_list, value=loras_list[0], label="LoRA")
|
151 |
+
refresh_btn = gr.Button("Refresh Dropdowns")
|
152 |
prompt = gr.Textbox(label="Prompt", elem_id="prompt_textbox")
|
153 |
generate_btn = gr.Button("Generate Image", elem_id="generate_button")
|
154 |
with gr.Accordion("Advanced", open=False, elem_id="advanced_accordion"):
|
|
|
161 |
with gr.Column(elem_id="output_column"):
|
162 |
output_gallery = gr.Gallery(label="Generated Images", height=480, scale=1, elem_id="output_gallery")
|
163 |
|
164 |
+
refresh_btn.click(refresh_dropdowns, outputs=[model_dropdown, lora_dropdown])
|
165 |
generate_btn.click(generate_images, inputs=[model_dropdown, lora_dropdown, prompt, negative_prompt, num_inference_steps, guidance_scale, height, width, num_images], outputs=output_gallery)
|
166 |
|
167 |
with gr.Tab("Download Custom Model"):
|