lichorosario commited on
Commit
52ce4a7
·
verified ·
1 Parent(s): 7abaf33

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 (custom_model, weight_name, 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,7 +137,7 @@ with gr.Blocks(css=css) as demo:
137
 
138
  submit_btn.click(
139
  fn = infer,
140
- inputs = [custom_model, weight_name, prompt_in, inf_steps, guidance_scale, width, height, seed, lora_weight],
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