Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix(router): fix api-inference deployment (#31)

This commit is contained in:
OlivierDehaene 2023-01-23 17:42:14 +01:00 committed by GitHub
parent ab2ad91da3
commit 5c01e2544c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
router/src/server.rs
View File

@ -195,8 +195,8 @@ pub async fn run(
// Create router // Create router
let app = Router::new() let app = Router::new()
.route("/", post(generate))
.route("/generate", post(generate)) .route("/generate", post(generate))
.layer(Extension(shared_state.clone()))
.route("/", get(health)) .route("/", get(health))
.route("/health", get(health)) .route("/health", get(health))
.layer(Extension(shared_state.clone())); .layer(Extension(shared_state.clone()));