fix request body

app.py

@@ -26,13 +26,16 @@ class Disease(BaseModel):
     name: str
     score: float
 
+class Symptoms(BaseModel):
+    query: str
+
 @app.get("/")
 def greet_json():
     return {"Hello": "World!"}
 
 @app.post("/", response_model=list[Disease])
-async def predict(query: str):
-    query_embedding = model.encode(query).astype('float')
+async def predict(symptoms: Symptoms):
+    query_embedding = model.encode(symptoms.query).astype('float')
     similarity_vectors = model.similarity(query_embedding, corpus)[0]
     scores, indicies = torch.topk(similarity_vectors, k=len(corpus))
     # print("Similarity Vector Shape: ", similarity_vectors.shape)