Update main.py

main.py

@@ -28,9 +28,10 @@ app.add_middleware(
 gr.mount_gradio_app(app, demo, path="/gradio")
 
 @app.post("/run")
-async def api_run(text: str):
+async def api_run(args: dict):
     try:
-        output = fn.fn_pke(text)
+        print(args['text'])
+        output = fn.fn_pke(args['text'])
 
         return {"result": output}
     except Exception as e: