diff --git a/xtask/src/service/mod.rs b/xtask/src/service/mod.rs index aaedcdc6..325e0e24 100644 --- a/xtask/src/service/mod.rs +++ b/xtask/src/service/mod.rs @@ -195,7 +195,9 @@ impl HyperService> for App { Err(e) => return Ok(error(Error::WrongJson(e))), }; let model_name = match &req.model { - serde_json::Value::String(s) => s.clone(), + serde_json::Value::String(s) => { + s.split("/").last().unwrap_or(s).to_string() + } _ => { return Ok(error(Error::ModelNotFound( "model field must be a string".to_string(),