mudaza commited on
Commit
17faa08
·
1 Parent(s): ffa5f5d

update code

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -33,12 +33,12 @@ async def predict(query: str):
33
  query_embedding = model.encode(query).astype('float')
34
  similarity_vectors = model.similarity(q, all_embeddings)
35
  scores, indicies = torch.topk(similarity_vectors, k=len(all_embeddings))
36
- id = df.iloc[indicies]
37
- id = df.drop_duplicates("label")
38
- scores = scores[id.index]
39
- diseases = label_encoder.inverse_transform(id.label.values)
40
- id = id.label.values
41
- diseases = [dict("id": value[0], "name": value[1], "score" : value[2]) for value in zip(id, diseases, scores)]
42
  return diseases
43
 
44
 
 
33
  query_embedding = model.encode(query).astype('float')
34
  similarity_vectors = model.similarity(q, all_embeddings)
35
  scores, indicies = torch.topk(similarity_vectors, k=len(all_embeddings))
36
+ id_ = df.iloc[indicies]
37
+ id_ = df.drop_duplicates("label")
38
+ scores = scores[id_.index]
39
+ diseases = label_encoder.inverse_transform(id_.label.values)
40
+ id_ = id_.label.values
41
+ diseases = [dict("id": value[0], "name": value[1], "score" : value[2]) for value in zip(_id, diseases, scores)]
42
  return diseases
43
 
44