more steps

app.py

@@ -23,7 +23,7 @@ if preferred_device == "cuda":
 
 inpainting_pipeline = AutoPipelineForInpainting.from_pretrained(
     "runwayml/stable-diffusion-inpainting",
-    revision="fp16",
+    variant="fp16",
     torch_dtype=preferred_dtype,
 ).to(preferred_device)
 
@@ -64,14 +64,14 @@ def app(img, prompt):
         image=img,
         mask_image=mask,
         strength=0.95,
-        num_inference_steps=20,
+        num_inference_steps=50,
     ).images[0]
     end_time = datetime.now().timestamp()
     draw = ImageDraw.Draw(overlay_img)
     # replace spaces with newlines after many words to line break prompt
     prompt = " ".join([prompt.split(" ")[i] if (i+1) % 5 else prompt.split(" ")[i] + "\n" for i in range(len(prompt.split(" ")))])
 
-    draw.text((50, 10), f"Old size: {old_size}\nTotal duration: {int(1000 * (end_time - start_time))}ms\nSegmentation {int(1000 * (mask_time - start_time))}ms / inpainting {int(1000 * (end_time - mask_time))} \n<{prompt}>", fill=(255, 255, 255))
+    draw.text((50, 10), f"Old size: {old_size}\nTotal duration: {int(1000 * (end_time - start_time))}ms\nSegmentation {int(1000 * (mask_time - start_time))}ms / inpainting {int(1000 * (end_time - mask_time))} \n<{prompt}>", fill=(123, 0, 123))
     return overlay_img
 
 #ideally: