update code

app.py

@@ -38,7 +38,10 @@ def greet_json():
 async def predict(query: str):
     query_embedding = model.encode(query).astype('float')
     similarity_vectors = model.similarity(query_embedding, corpus)
+    print("Similarity Vector Shape: ", similarity_vectors.shape)
     scores, indicies = torch.topk(similarity_vectors, k=len(corpus))
+    print("Scores Shape: ", scores.shape)
+    print("Indicies Shape: ", indicies.shape)
     id_ = df.iloc[indicies]
     id_ = df.drop_duplicates("label")
     scores = scores[id_.index]