mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2025-01-06 10:25:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

cleanups

Browse Source
This commit is contained in:
Abubakar Abid 2022-03-04 00:06:30 -05:00
parent 9765c0a536
commit 5cf8bcbc01
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
gradio/interface.py
View File

@ -202,7 +202,8 @@ class Interface(Launchable):
].index(True)
self.state_return_index = state_return_index
else:
raise ValueError("Exactly one input and one output component must be State")
raise ValueError("For a stateful interface, there must be exactly one State"
" input component and one State output component.")
if (
interpretation is None
@ -556,11 +557,7 @@ class Interface(Launchable):
else:
return predictions
def process_api(
self,
data: Dict[str, Any],
username: str = None,
) -> Dict[str, Any]:
def process_api(self, data: Dict[str, Any], username: str = None) -> Dict[str, Any]:
flag_index = None
if data.get("example_id") is not None:
example_id = data["example_id"]
@ -585,6 +582,7 @@ class Interface(Launchable):
)
if self.stateful:
updated_state = prediction[self.state_return_index]
prediction[self.state_return_index] = None
else:
updated_state = None