From 5c01e2544c0a2a91b2e9aff8a24c18da9b485a37 Mon Sep 17 00:00:00 2001
From: OlivierDehaene <olivier@huggingface.co>
Date: Mon, 23 Jan 2023 17:42:14 +0100
Subject: [PATCH] fix(router): fix api-inference deployment (#31)

---
 router/src/server.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/router/src/server.rs b/router/src/server.rs
index db6c369..623dd07 100644
--- a/router/src/server.rs
+++ b/router/src/server.rs
@@ -195,8 +195,8 @@ pub async fn run(
 
     // Create router
     let app = Router::new()
+        .route("/", post(generate))
         .route("/generate", post(generate))
-        .layer(Extension(shared_state.clone()))
         .route("/", get(health))
         .route("/health", get(health))
         .layer(Extension(shared_state.clone()));