Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -57,8 +57,10 @@ def load_model(custom_model):
|
|
57 |
|
58 |
print(f"Safetensors available: {sfts_available_files}")
|
59 |
|
60 |
-
return gr.update(choices=sfts_available_files, value=sfts_available_files[0], visible=True), gr.update(value=instance_prompt, visible=True)
|
61 |
|
|
|
|
|
62 |
|
63 |
@spaces.GPU
|
64 |
def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):
|
@@ -187,52 +189,67 @@ with gr.Blocks(css=css) as demo:
|
|
187 |
)
|
188 |
with gr.Column():
|
189 |
load_model_btn = gr.Button("Load my model")
|
|
|
|
|
|
|
|
|
190 |
trigger_word = gr.Textbox(label="Trigger word", interactive=False, visible=False)
|
191 |
|
192 |
-
prompt_in = gr.Textbox(
|
193 |
-
|
194 |
-
|
195 |
-
|
196 |
-
minimum=12,
|
197 |
-
maximum=50,
|
198 |
-
step=1,
|
199 |
-
value=25
|
200 |
-
)
|
201 |
-
guidance_scale = gr.Slider(
|
202 |
-
label="Guidance scale",
|
203 |
-
minimum=0.0,
|
204 |
-
maximum=50.0,
|
205 |
-
step=0.1,
|
206 |
-
value=7.5
|
207 |
-
)
|
208 |
|
209 |
-
with gr.
|
210 |
-
|
211 |
-
|
212 |
-
|
213 |
-
|
214 |
-
|
215 |
-
|
216 |
-
|
217 |
-
|
218 |
-
|
219 |
-
|
220 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
221 |
)
|
222 |
-
lora_weight = gr.Slider(
|
223 |
-
label="LoRa weigth",
|
224 |
-
minimum=0.0,
|
225 |
-
maximum=1.0,
|
226 |
-
step=0.01,
|
227 |
-
value=0.9
|
228 |
-
)
|
229 |
submit_btn = gr.Button("Submit")
|
230 |
image_out = gr.Image(label="Image output")
|
231 |
|
|
|
|
|
|
|
|
|
|
|
|
|
232 |
load_model_btn.click(
|
233 |
fn = load_model,
|
234 |
inputs=[custom_model],
|
235 |
-
outputs = [weight_name, trigger_word],
|
236 |
queue = False
|
237 |
)
|
238 |
submit_btn.click(
|
|
|
57 |
|
58 |
print(f"Safetensors available: {sfts_available_files}")
|
59 |
|
60 |
+
return "Model Ready", gr.update(choices=sfts_available_files, value=sfts_available_files[0], visible=True), gr.update(value=instance_prompt, visible=True)
|
61 |
|
62 |
+
def custom_model_changed(custom_model):
|
63 |
+
return "Model changed, you must reload before re-run"
|
64 |
|
65 |
@spaces.GPU
|
66 |
def infer (custom_model, weight_name, prompt, inf_steps, guidance_scale, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):
|
|
|
189 |
)
|
190 |
with gr.Column():
|
191 |
load_model_btn = gr.Button("Load my model")
|
192 |
+
model_status = gr.Textbox(
|
193 |
+
label = "model status",
|
194 |
+
show_label = False
|
195 |
+
)
|
196 |
trigger_word = gr.Textbox(label="Trigger word", interactive=False, visible=False)
|
197 |
|
198 |
+
prompt_in = gr.Textbox(
|
199 |
+
label="Your Prompt",
|
200 |
+
info = "Dont' forget to include your trigger word if necessary"
|
201 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
202 |
|
203 |
+
with gr.Accordion("Advanced Settings", open=False):
|
204 |
+
with gr.Row():
|
205 |
+
inf_steps = gr.Slider(
|
206 |
+
label="Inference steps",
|
207 |
+
minimum=12,
|
208 |
+
maximum=50,
|
209 |
+
step=1,
|
210 |
+
value=25
|
211 |
+
)
|
212 |
+
guidance_scale = gr.Slider(
|
213 |
+
label="Guidance scale",
|
214 |
+
minimum=0.0,
|
215 |
+
maximum=50.0,
|
216 |
+
step=0.1,
|
217 |
+
value=7.5
|
218 |
+
)
|
219 |
+
|
220 |
+
with gr.Row():
|
221 |
+
seed = gr.Slider(
|
222 |
+
label="Seed",
|
223 |
+
info = "-1 denotes a random seed",
|
224 |
+
minimum=-1,
|
225 |
+
maximum=423538377342,
|
226 |
+
step=1,
|
227 |
+
value=-1
|
228 |
+
)
|
229 |
+
last_used_seed = gr.Number(
|
230 |
+
label = "Last used seed",
|
231 |
+
info = "the seed used in the last generation",
|
232 |
+
)
|
233 |
+
lora_weight = gr.Slider(
|
234 |
+
label="LoRa weigth",
|
235 |
+
minimum=0.0,
|
236 |
+
maximum=1.0,
|
237 |
+
step=0.01,
|
238 |
+
value=0.9
|
239 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
240 |
submit_btn = gr.Button("Submit")
|
241 |
image_out = gr.Image(label="Image output")
|
242 |
|
243 |
+
custom_model.blur(
|
244 |
+
fn=custom_model_changed,
|
245 |
+
inputs = [custom_model],
|
246 |
+
outputs = [model_status],
|
247 |
+
queue = False
|
248 |
+
)
|
249 |
load_model_btn.click(
|
250 |
fn = load_model,
|
251 |
inputs=[custom_model],
|
252 |
+
outputs = [model_status, weight_name, trigger_word],
|
253 |
queue = False
|
254 |
)
|
255 |
submit_btn.click(
|