aka7774 commited on
Commit
b16615c
1 Parent(s): 45e2951

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -67,5 +67,5 @@ async def api_embedding(args: dict):
67
 
68
  @app.post("/search")
69
  async def api_search(args: dict):
70
- result = fn.search(args['name'], args['query'], args['num'] if 'num' in args else 3)
71
  return {'result': result}
 
67
 
68
  @app.post("/search")
69
  async def api_search(args: dict):
70
+ result = fn.search(args['name'], args['query'], int(args['num']) if 'num' in args else 3)
71
  return {'result': result}