Merge pull request #10 from tomseimandi/master

labenech · web-flow · commit ef74816619e9 · 2022-04-27T17:07:33.000+02:00
Keep order of input for label output
diff --git a/app/main.py b/app/main.py
@@ -63,7 +63,7 @@ async def predict_label(q: List[str] = Query(..., title="query string", descript
 
     for nomenclature in n:
         output[nomenclature] = {}
-        descriptions = list(set(q))
+        descriptions = sorted(set(q), key=q.index)
         preprocessed_descriptions = [preprocess_text(description) for description in descriptions]
         preds = predict_using_model(preprocessed_descriptions, model=models[nomenclature], k=k)
         if v:
