From 087b4c27218c118dc912d55410a626a15a93817b Mon Sep 17 00:00:00 2001
From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com>
Date: Mon, 23 Jan 2023 17:41:42 +0100
Subject: [PATCH] fix(router): fix api-inference deployment

---
 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 db6c369b..623dd07c 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()));