Fix sort ordering

backend/inference.py

@@ -20,18 +20,18 @@ def text_similarity(anchor: str, inputs: List[str], model_name: str, model_dict:
     # Creating embeddings
     if hasattr(model, 'encode'):
         anchor_emb = model.encode(anchor)[None, :]
-        inputs_emb = model.encode([input for input in inputs])
+        inputs_emb = model.encode(inputs)
     else:
         assert len(model) == 2
         anchor_emb = model[0].encode(anchor)[None, :]
-        inputs_emb = model[1].encode([input for input in inputs])
+        inputs_emb = model[1].encode(inputs)
 
     # Obtaining similarity
     similarity = list(jnp.squeeze(cos_sim(anchor_emb, inputs_emb)))
 
     # Returning a Pandas' dataframe
-    d = {'inputs': [input for input in inputs],
+    d = {'inputs': inputs,
          'score': [round(similarity[i], 3) for i in range(len(similarity))]}
     df = pd.DataFrame(d, columns=['inputs', 'score'])
 
-    return df.sort_values('score', ascending=False)
+    return df