Spaces:
Runtime error
Runtime error
pohjie
commited on
Commit
β’
09bdd6d
1
Parent(s):
7cea2de
Add symptom list to return result
Browse files
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 |
-
|
48 |
-
|
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(
|