Update handler.py
Browse files- handler.py +6 -6
handler.py
CHANGED
|
@@ -238,9 +238,9 @@ class EndpointHandler():
|
|
| 238 |
A :obj:`dict`:. base64 encoded image
|
| 239 |
"""
|
| 240 |
# deserialize incomin request
|
| 241 |
-
inputs = data.pop("inputs", data)
|
| 242 |
# parameters = data.pop("parameters", None)
|
| 243 |
-
model_id = data.pop("model_id", None)
|
| 244 |
|
| 245 |
# check if model_id is in the list of models
|
| 246 |
# if model_id is None or model_id not in multi_model_list:
|
|
@@ -256,9 +256,9 @@ class EndpointHandler():
|
|
| 256 |
|
| 257 |
|
| 258 |
|
| 259 |
-
|
| 260 |
-
print("Logs post: self.path",self.path)
|
| 261 |
print("Logs post: task is ", data)
|
|
|
|
| 262 |
inputs = data.pop("inputs", data)
|
| 263 |
parameters = data.pop("parameters", None)
|
| 264 |
model_id = data.pop("model_id", None)
|
|
@@ -287,8 +287,8 @@ class EndpointHandler():
|
|
| 287 |
raise Exception("Invalid task type")
|
| 288 |
except Exception as e:
|
| 289 |
print(f"Error: {e}")
|
| 290 |
-
slack.error_alert(
|
| 291 |
-
return
|
| 292 |
|
| 293 |
# inputs = data.pop("inputs", data)
|
| 294 |
|
|
|
|
| 238 |
A :obj:`dict`:. base64 encoded image
|
| 239 |
"""
|
| 240 |
# deserialize incomin request
|
| 241 |
+
# inputs = data.pop("inputs", data)
|
| 242 |
# parameters = data.pop("parameters", None)
|
| 243 |
+
# model_id = data.pop("model_id", None)
|
| 244 |
|
| 245 |
# check if model_id is in the list of models
|
| 246 |
# if model_id is None or model_id not in multi_model_list:
|
|
|
|
| 256 |
|
| 257 |
|
| 258 |
|
| 259 |
+
print("Logs post: init")
|
|
|
|
| 260 |
print("Logs post: task is ", data)
|
| 261 |
+
|
| 262 |
inputs = data.pop("inputs", data)
|
| 263 |
parameters = data.pop("parameters", None)
|
| 264 |
model_id = data.pop("model_id", None)
|
|
|
|
| 287 |
raise Exception("Invalid task type")
|
| 288 |
except Exception as e:
|
| 289 |
print(f"Error: {e}")
|
| 290 |
+
slack.error_alert(data, e)
|
| 291 |
+
return { "error": e, "data":data, "task_type":task_type, "self.multi_text2image_model[model_id]":self.multi_text2image_model[model_id] }
|
| 292 |
|
| 293 |
# inputs = data.pop("inputs", data)
|
| 294 |
|