ka1kuk commited on
Commit
f29ef8b
·
1 Parent(s): 93962d9

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -56,7 +56,7 @@ async def searches(request: Request):
56
  response = agent_chain.run(input=prompt)
57
  return response
58
 
59
- @app.route("/imagen", methods=["POST"])
60
  async def generate_image(request: Request):
61
  data = await request.json()
62
  prompt = data['prompt']
@@ -75,7 +75,7 @@ class User(BaseModel):
75
  seed: int
76
  neg: str | None = None
77
 
78
- @app.route("/test", methods=["POST"])
79
  async def test(request: User):
80
 
81
  return request
 
56
  response = agent_chain.run(input=prompt)
57
  return response
58
 
59
+ @app.post("/imagen")
60
  async def generate_image(request: Request):
61
  data = await request.json()
62
  prompt = data['prompt']
 
75
  seed: int
76
  neg: str | None = None
77
 
78
+ @app.post("/test")
79
  async def test(request: User):
80
 
81
  return request