Commit ·
186d8fb
1
Parent(s): c70f9b8
Increase number of steps
Browse files
app.py
CHANGED
|
@@ -229,7 +229,7 @@ async def inpaint_websocket_endpoint(websocket: WebSocket):
|
|
| 229 |
mask = crop_image(resize_image(image, 512, 512))
|
| 230 |
|
| 231 |
pipe = app.package.get('pipe')
|
| 232 |
-
result = pipe(prompt=prompt, image=image, mask_image=mask, num_inference_steps=
|
| 233 |
images = result['images']
|
| 234 |
|
| 235 |
# Convert the result to base64 and send the json back
|
|
|
|
| 229 |
mask = crop_image(resize_image(image, 512, 512))
|
| 230 |
|
| 231 |
pipe = app.package.get('pipe')
|
| 232 |
+
result = pipe(prompt=prompt, image=image, mask_image=mask, num_inference_steps=20, num_images_per_prompt=1)
|
| 233 |
images = result['images']
|
| 234 |
|
| 235 |
# Convert the result to base64 and send the json back
|