shreyankisiri commited on
Commit
82d4b08
·
verified ·
1 Parent(s): 6bf4fee

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -9
main.py CHANGED
@@ -56,15 +56,10 @@ def find_similar_courses(query_text):
56
  async def health():
57
  return {"status": "OK"}
58
 
59
- @app.get("/search")
60
- async def search(query: str):
61
- top_courses = find_similar_courses(query)
62
-
63
- l = []
64
- for course in top_courses:
65
- l.append(course["course_id"])
66
-
67
- return l
68
 
69
  if __name__ == "__main__":
70
  import uvicorn
 
56
  async def health():
57
  return {"status": "OK"}
58
 
59
+ @app.post("/search")
60
+ async def search(request: SearchRequest):
61
+ top_courses = find_similar_courses(request.query)
62
+ return [course["course_id"] for course in top_courses]
 
 
 
 
 
63
 
64
  if __name__ == "__main__":
65
  import uvicorn