abyildirim commited on
Commit
fee061d
·
1 Parent(s): 88022c3

timestep decreased

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def inference(image: np.ndarray, instruction: str, center_crop: bool):
29
  raise gr.Error("Instruction should start with 'Remove the' !")
30
  image = Image.fromarray(image)
31
  cropped_image, image = utils.preprocess_image(image, center_crop=center_crop)
32
- output_image = MODEL.inpaint(image, instruction, num_steps=50, device="cuda", return_pil=True, seed=0)
33
  return cropped_image, output_image
34
 
35
  if __name__ == "__main__":
 
29
  raise gr.Error("Instruction should start with 'Remove the' !")
30
  image = Image.fromarray(image)
31
  cropped_image, image = utils.preprocess_image(image, center_crop=center_crop)
32
+ output_image = MODEL.inpaint(image, instruction, num_steps=10, device="cuda", return_pil=True, seed=0)
33
  return cropped_image, output_image
34
 
35
  if __name__ == "__main__":