Update app.py

app.py

@@ -34,7 +34,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 pipe = StableDiffusionInpaintPipeline.from_pretrained(
     "runwayml/stable-diffusion-inpainting",
     revision="fp16",
-    #torch_dtype=torch.float16,
+    torch_dtype=torch.float16,
 ).to(device)
 
 def predict(image, prompt):
@@ -57,6 +57,7 @@ def predict(image, prompt):
     image=image,
     mask_image=mask_image,
     guidance_scale=5,
+    generator=torch.Generator(device="cuda").manual_seed(0),
     num_images_per_prompt=1,
   ).images