update code

app.py

@@ -43,11 +43,9 @@ async def predict(query: str):
     print("Scores Shape: ", scores.shape)
     print("Indicies Shape: ", indicies.shape)
     id_ = df.iloc[indicies]
-    id_ = df.drop_duplicates("label")
+    id_ = id_.drop_duplicates("label")
     scores = scores[id_.index]
     diseases = label_encoder.inverse_transform(id_.label.values)
     id_ = id_.label.values
     diseases = [dict({"id": value[0], "name": value[1], "score" : value[2]}) for value in zip(id_, diseases, scores)]
-    return diseases
-
-
+    return diseases