jayparmr commited on
Commit
61826f1
·
1 Parent(s): 7916b03

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -243,7 +243,7 @@ class EndpointHandler():
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 self.multi_model_list:
247
  raise ValueError(f"model_id: {model_id} is not valid. Available models are: {list(self.multi_model.keys())}")
248
 
249
  # # pass inputs with all kwargs in data
 
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:
247
  raise ValueError(f"model_id: {model_id} is not valid. Available models are: {list(self.multi_model.keys())}")
248
 
249
  # # pass inputs with all kwargs in data