fixed text worker

LAION-AI · Mar 11, 2023 · 17b6145 · 17b6145
1 parent 50b81f1
commit 17b6145
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inference/text-client/__main__.py b/inference/text-client/__main__.py
@@ -70,7 +70,7 @@ def main(backend_url: str = "http://127.0.0.1:8000"):
                         raise
                     if error := data.get("error"):
                         raise Exception(error)
-                    print(data["token"]["text"], end="", flush=True)
+                    print(data["text"], end="", flush=True)
                 print()
                 parent_id = message_id
         except typer.Abort: