Update handler.py

handler.py

@@ -279,21 +279,21 @@ class EndpointHandler():
         try:
             task_type = task.get_type()
             print("logs post: self.multi_text2image_model[model_id]", self.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[model_id])
-            # # elif task_type == TaskType.IMAGE_TO_IMAGE:
-            # #     return img2img(task)
-            # # elif task_type == TaskType.CANNY:
-            # #     return canny(task)
-            # # elif task_type == TaskType.POSE:
-            # #     return pose(task)
-            # else:
-            #     raise Exception("Invalid task type")
+            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"])
+            # elif task_type == TaskType.IMAGE_TO_IMAGE:
+            #     return img2img(task)
+            # elif task_type == TaskType.CANNY:
+            #     return canny(task)
+            # elif task_type == TaskType.POSE:
+            #     return pose(task)
+            else:
+                raise Exception("Invalid task type")
         except Exception as e:
             print(f"Error: {e}")
             slack.error_alert(data, e)