From a1bc9378a051591de9c272d3e1ba7b09f64782c3 Mon Sep 17 00:00:00 2001
From: Abubakar Abid <abubakar@huggingface.co>
Date: Tue, 8 Mar 2022 19:45:25 -0600
Subject: [PATCH] fixed predict body issue

---
 gradio/blocks.py | 7 ++++---
 gradio/routes.py | 1 +
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/gradio/blocks.py b/gradio/blocks.py
index ad87aacd09..ddb8424f55 100644
--- a/gradio/blocks.py
+++ b/gradio/blocks.py
@@ -3,6 +3,7 @@ from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional, Tuple
 from gradio import utils
 from gradio.context import Context
 from gradio.launchable import Launchable
+from gradio.routes import PredictBody
 
 
 class Block:
@@ -81,9 +82,9 @@ class Blocks(Launchable, BlockContext):
         self.fns = []
         self.dependencies = []
 
-    def process_api(self, data: Dict[str, Any], username: str = None) -> Dict[str, Any]:
-        raw_input = data["data"]
-        fn_index = data["fn_index"]
+    def process_api(self, data: PredictBody, username: str = None) -> Dict[str, Any]:
+        raw_input = data.data
+        fn_index = data.fn_index
         fn = self.fns[fn_index]
         dependency = self.dependencies[fn_index]
 
diff --git a/gradio/routes.py b/gradio/routes.py
index 159ecfa726..36385ce4e1 100644
--- a/gradio/routes.py
+++ b/gradio/routes.py
@@ -71,6 +71,7 @@ class PredictBody(BaseModel):
     example_id: Optional[int]
     data: List[Any]
     state: Optional[Any]
+    fn_index: Optional[int]
 
 
 class FlagData(BaseModel):