Update endpoints.py

endpoints.py

@@ -5,12 +5,19 @@ from huggingface_hub import login
 from config import settings
 import torch
 # from diffusers import StableDiffusionPipeline
-from diffusers import StableDiffusionInpaintPipeline
+# from diffusers import StableDiffusionInpaintPipeline
 
-# pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16)
-pipe = StableDiffusionInpaintPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-2-inpainting",
-    torch_dtype=torch.float16,
+# # pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16)
+# pipe = StableDiffusionInpaintPipeline.from_pretrained(
+#     "stabilityai/stable-diffusion-2-inpainting",
+#     torch_dtype=torch.float16,
+# )
+
+from diffusers import AutoPipelineForInpainting
+from diffusers.utils import load_image, make_image_grid
+
+pipee = AutoPipelineForInpainting.from_pretrained(
+    "kandinsky-community/kandinsky-2-2-decoder-inpaint", torch_dtype=torch.float16
 )
 
 login(settings.huggingface_key)