makakwastaken commited on
Commit
186d8fb
·
1 Parent(s): c70f9b8

Increase number of steps

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=10, num_images_per_prompt=1)
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