diff --git a/ai_server.py b/ai_server.py index 2f0a54e..6baf056 100644 --- a/ai_server.py +++ b/ai_server.py @@ -73,11 +73,11 @@ return {"message" : "Hello,World"} @app.get("/render") -async def render(): +async def render(prompt: str = "", outputdir: str = "output/default"): # キューに情報を追加 imgOrder = {} - imgOrder["prompt"] = "cute cat" - imgOrder["outputdir"] = "output/cutecat01" + imgOrder["prompt"] = prompt + imgOrder["outputdir"] = outputdir q.put(imgOrder) return {"message" : "render"}