Rooni commited on
Commit
3899612
·
verified ·
1 Parent(s): 8acf7e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -9
app.py CHANGED
@@ -119,16 +119,19 @@ def query(prompt, model, is_negative=False, steps=30, cfg_scale=7, sampler="DPM+
119
 
120
  payload = {
121
  "inputs": prompt,
122
- "width": width,
123
- "height": height,
124
- "is_negative": is_negative,
125
- "steps": steps,
126
- "cfg_scale": cfg_scale,
127
- "seed": seed if seed != -1 else random.randint(1, 999999),
128
- "guidance_scale": cfg_scale,
129
- "num_inference_steps": steps,
130
- "negative_prompt": is_negative
 
 
131
  }
 
132
 
133
  response = requests.post(f"{api_base}{API_URL}", headers=headers, json=payload, timeout=timeout)
134
  if response.status_code != 200:
 
119
 
120
  payload = {
121
  "inputs": prompt,
122
+ "parameters": {
123
+ "target_size": {
124
+ "width": width,
125
+ "height": height,
126
+ },
127
+ "guidance_scale": cfg_scale,
128
+ "seed": seed if seed != -1 else random.randint(1, 999999),
129
+ "guidance_scale": cfg_scale,
130
+ "num_inference_steps": steps,
131
+ "negative_prompt": is_negative,
132
+ "scheduler": sampler
133
  }
134
+ }
135
 
136
  response = requests.post(f"{api_base}{API_URL}", headers=headers, json=payload, timeout=timeout)
137
  if response.status_code != 200: