Update handler.py
Browse files- 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
|
| 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
|