Spaces:
Runtime error
Runtime error
Commit
·
fee061d
1
Parent(s):
88022c3
timestep decreased
Browse files
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=
|
| 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__":
|