Update handler.py

handler.py

@@ -192,6 +192,10 @@ multi_model_list = [
     {"model_id": "jcplus/stable-diffusion-v1-5"},
 ]
 
+multi_controlnet_model={}
+multi_text2image_model={}
+multi_image2image_model={}
+
 class EndpointHandler():
     def __init__(self, path=""):
         # load the optimized model
@@ -201,9 +205,7 @@ class EndpointHandler():
 
         lora_style.load(path)
         
-        self.multi_controlnet_model={}
-        self.multi_text2image_model={}
-        self.multi_image2image_model={}
+        
         self.path = path
         
         for model in multi_model_list:
@@ -215,26 +217,26 @@ class EndpointHandler():
             controlnet = ControlNet()
             img2img_pipe = Img2Img()
             text2img_pipe = Text2Img()
-            self.multi_controlnet_model[model["model_id"]] = controlnet;
+            multi_controlnet_model[model["model_id"]] = controlnet;
             controlnet.load(model["model_id"])
             
             
-            self.multi_text2image_model[model["model_id"]] = text2img_pipe;
+            multi_text2image_model[model["model_id"]] = text2img_pipe;
             text2img_pipe.load( model["model_id"])
             
-            self.multi_image2image_model[model["model_id"]] = img2img_pipe;
+            multi_image2image_model[model["model_id"]] = img2img_pipe;
             img2img_pipe.load( model["model_id"])
             
             print(" Logs: model[model_id]", model["model_id"])
-            print("Logs: multimodel controlnet pipelines are",  self.multi_controlnet_model[model["model_id"]])    
-            print("Logs: multimodel text2img pipelines are",  self.multi_text2image_model[model["model_id"]]) 
-            print("Logs: multimodel imgtoimage pipelines are",  self.multi_image2image_model[model["model_id"]]) 
+            print("Logs: multimodel controlnet pipelines are",  multi_controlnet_model[model["model_id"]])    
+            print("Logs: multimodel text2img pipelines are",  multi_text2image_model[model["model_id"]]) 
+            print("Logs: multimodel imgtoimage pipelines are",  multi_image2image_model[model["model_id"]]) 
         # controlnet.load(path)
         # text2img_pipe.load(path)
         # img2img_pipe.load(path)
-        print("Logs: self.multi_image2image_model",self.multi_image2image_model)
-        print("Logs: self.multi_text2image_model",self.multi_text2image_model)
-        print("Logs: self.multi_controlnet_model",self.multi_controlnet_model)
+        print("Logs: self.multi_image2image_model",multi_image2image_model)
+        print("Logs: self.multi_text2image_model",multi_text2image_model)
+        print("Logs: self.multi_controlnet_model",multi_controlnet_model)
         print("Logs: model loaded ....")
         
 
@@ -280,14 +282,14 @@ class EndpointHandler():
         
         try:
             task_type = task.get_type()
-            print("logs post: self.multi_text2image_model[model_id]", self.multi_text2image_model) 
+            print("logs post: self.multi_text2image_model[model_id]", multi_text2image_model) 
             if task_type == TaskType.TEXT_TO_IMAGE:
                 #  character sheet
                 if "character sheet" in task.get_prompt().lower():
                     print("pose is here")
                     # return pose(task, s3_outkey="", poses=pickPoses())
                 else:
-                    return text2img(task, self.multi_text2image_model["jayparmr/icbinp"])
+                    return text2img(task, multi_text2image_model["jayparmr/icbinp"])
             # elif task_type == TaskType.IMAGE_TO_IMAGE:
             #     return img2img(task)
             # elif task_type == TaskType.CANNY: