update code

app.py

@@ -37,7 +37,7 @@ def greet_json():
 @app.post("/", response_model=list[Disease])
 async def predict(query: str):
     query_embedding = model.encode(query).astype('float')
-    similarity_vectors = model.similarity(query_embedding, corpus)
+    similarity_vectors = model.similarity(query_embedding, corpus)[0]
     print("Similarity Vector Shape: ", similarity_vectors.shape)
     scores, indicies = torch.topk(similarity_vectors, k=len(corpus))
     print("Scores Shape: ", scores.shape)