Update app.py
Browse files
app.py
CHANGED
|
@@ -131,13 +131,13 @@ def generate(input=DEFAULT_INPUT, filter_input="", negative_input=DEFAULT_NEGATI
|
|
| 131 |
parameters = {
|
| 132 |
"prompt": input,
|
| 133 |
"negative_prompt": filter_input + negative_input,
|
| 134 |
-
|
| 135 |
-
|
| 136 |
"num_inference_steps": steps,
|
| 137 |
"guidance_scale": guidance,
|
| 138 |
-
|
| 139 |
"generator": torch.Generator().manual_seed(seed),
|
| 140 |
-
|
| 141 |
}
|
| 142 |
|
| 143 |
images = repo(**parameters).images
|
|
|
|
| 131 |
parameters = {
|
| 132 |
"prompt": input,
|
| 133 |
"negative_prompt": filter_input + negative_input,
|
| 134 |
+
"height": height,
|
| 135 |
+
"width": width,
|
| 136 |
"num_inference_steps": steps,
|
| 137 |
"guidance_scale": guidance,
|
| 138 |
+
"num_images_per_prompt": number,
|
| 139 |
"generator": torch.Generator().manual_seed(seed),
|
| 140 |
+
"output_type":"pil",
|
| 141 |
}
|
| 142 |
|
| 143 |
images = repo(**parameters).images
|