pohjie commited on
Commit
09bdd6d
β€’
1 Parent(s): 7cea2de

Add symptom list to return result

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -44,9 +44,14 @@ def get_top_n_diagnoses(case_note):
44
  count += 1
45
  score = cosine_similarity(criterion["embedding"], case_note_emb)
46
  sum_score += score
47
- criteria_scores.append(
48
- {"description": criterion["description"], "score": float(score)}
49
- )
 
 
 
 
 
50
 
51
  sum_score /= count
52
  diagnosis_list.append(
 
44
  count += 1
45
  score = cosine_similarity(criterion["embedding"], case_note_emb)
46
  sum_score += score
47
+
48
+ criterion_data = {
49
+ "description": criterion["description"],
50
+ "score": float(score),
51
+ }
52
+ if "symptom_list" in criterion:
53
+ criterion_data["symptom_list"] = criterion["symptom_list"]
54
+ criteria_scores.append(criterion_data)
55
 
56
  sum_score /= count
57
  diagnosis_list.append(