Update handler.py

handler.py

@@ -3,16 +3,16 @@ from PIL import Image
 from io import BytesIO
 import torch
 import base64
-from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
+from diffusers import StableDiffusionImg2ImgPipeline
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
 class EndpointHandler():
     def __init__(self, path=""):
-        model_id = "timbrooks/instruct-pix2pix"
-        self.pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32, safety_checker=None)
+        model_id = "hafsa000/interior-design"
+        self.pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32, safety_checker=None)
         self.pipe.to(device)
-        self.pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(self.pipe.scheduler.config)
+        self.pipe.scheduler = PNDMScheduler.from_config(self.pipe.scheduler.config)
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """