Spaces:
Sleeping
Sleeping
lichorosario
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ from gradio_client import Client
|
|
11 |
custom_model="lichorosario/dott_remastered_style_lora_sdxl",
|
12 |
weight_name="dott_style.safetensors",
|
13 |
|
14 |
-
def infer (
|
15 |
client = Client("fffiloni/sd-xl-custom-model")
|
16 |
result = client.predict(
|
17 |
custom_model=custom_model,
|
@@ -137,7 +137,7 @@ with gr.Blocks(css=css) as demo:
|
|
137 |
|
138 |
submit_btn.click(
|
139 |
fn = infer,
|
140 |
-
inputs = [
|
141 |
outputs = [image_out, last_used_seed]
|
142 |
)
|
143 |
|
|
|
11 |
custom_model="lichorosario/dott_remastered_style_lora_sdxl",
|
12 |
weight_name="dott_style.safetensors",
|
13 |
|
14 |
+
def infer (prompt, inf_steps, guidance_scale, width, height, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):
|
15 |
client = Client("fffiloni/sd-xl-custom-model")
|
16 |
result = client.predict(
|
17 |
custom_model=custom_model,
|
|
|
137 |
|
138 |
submit_btn.click(
|
139 |
fn = infer,
|
140 |
+
inputs = [prompt_in, inf_steps, guidance_scale, width, height, seed, lora_weight],
|
141 |
outputs = [image_out, last_used_seed]
|
142 |
)
|
143 |
|