Update endpoints.py

endpoints.py

@@ -4,21 +4,8 @@ from routers import training
 from huggingface_hub import login
 from config import settings
 import torch
-# from diffusers import StableDiffusionPipeline
-# 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,
-# )
-
-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
-)
+from diffusers import StableDiffusionPipeline
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4",  torch_dtype=torch.float16)
 
 login(settings.huggingface_key)