update code
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ async def predict(query: str):
|
|
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(
|
42 |
return diseases
|
43 |
|
44 |
|
|
|
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 |
|