Sfoglia il codice sorgente

Change fn_index in api_example_stream (#904)

Andy Salerno 2 anni fa
parent
commit
3850f13624
1 ha cambiato i file con 7 aggiunte e 2 eliminazioni
  1. 7 2
      api-example-stream.py

+ 7 - 2
api-example-stream.py

@@ -12,6 +12,11 @@ import string
 
 import websockets
 
+# Note, Gradio may pick a different fn value as the definition of the Gradio app changes.
+# You can always launch the web UI and inspect the websocket stream using your browser's dev tools
+# to determine what value Gradio expects here.
+GRADIO_FN = 8
+
 
 def random_hash():
     letters = string.ascii_lowercase + string.digits
@@ -47,14 +52,14 @@ async def run(context):
                 case "send_hash":
                     await websocket.send(json.dumps({
                         "session_hash": session,
-                        "fn_index": 12
+                        "fn_index": GRADIO_FN
                     }))
                 case "estimation":
                     pass
                 case "send_data":
                     await websocket.send(json.dumps({
                         "session_hash": session,
-                        "fn_index": 12,
+                        "fn_index": GRADIO_FN,
                         "data": [
                             payload
                         ]