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[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")
         except Exception as e:
             print(f"Error: {e}")
             slack.error_alert(data, e)