Update main.py

main.py

@@ -56,7 +56,7 @@ async def searches(request: Request):
     response = agent_chain.run(input=prompt)
     return response
 
-@app.route("/imagen", methods=["POST"])
+@app.post("/imagen")
 async def generate_image(request: Request):
     data = await request.json()
     prompt = data['prompt']
@@ -75,7 +75,7 @@ class User(BaseModel):
     seed: int
     neg: str | None = None
 
-@app.route("/test", methods=["POST"])
+@app.post("/test")
 async def test(request: User):
     
     return request