jayparmr commited on
Commit
286a4d3
·
1 Parent(s): 874a8f4

Update handler.py

Browse files
Files changed (1) hide show
  1. 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(task, e)
291
- return None
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