From c79fc69e95fbe2aac85ccd414b92b7b3a425bd81 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 19 Mar 2023 10:36:57 -0300
Subject: [PATCH] Fix the API example with streaming #417

---
 api-example-stream.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/api-example-stream.py b/api-example-stream.py
index add1df4..055d605 100644
--- a/api-example-stream.py
+++ b/api-example-stream.py
@@ -44,14 +44,14 @@ async def run(context):
                 case "send_hash":
                     await websocket.send(json.dumps({
                         "session_hash": session,
-                        "fn_index": 7
+                        "fn_index": 9
                     }))
                 case "estimation":
                     pass
                 case "send_data":
                     await websocket.send(json.dumps({
                         "session_hash": session,
-                        "fn_index": 7,
+                        "fn_index": 9,
                         "data": [
                             context,
                             params['max_new_tokens'],