Spaces:
Running
Running
meghanaraok
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -166,6 +166,7 @@ def predict_icd(text_input, model_name, label_count):
|
|
166 |
label_dict = label_dict_df.set_index('icd9_code')['long_title'].to_dict()
|
167 |
predicted_labels_float = [float(label) for label in predicted_labels]
|
168 |
predicted_labels_with_titles = [(label, label_dict.get(str(label), "Not Found")) for label in predicted_labels_float]
|
|
|
169 |
|
170 |
|
171 |
html_output ="<h2>ICD Codes</h2>"
|
|
|
166 |
label_dict = label_dict_df.set_index('icd9_code')['long_title'].to_dict()
|
167 |
predicted_labels_float = [float(label) for label in predicted_labels]
|
168 |
predicted_labels_with_titles = [(label, label_dict.get(str(label), "Not Found")) for label in predicted_labels_float]
|
169 |
+
print("Keys in label_probabilities:", label_probabilities.keys())
|
170 |
|
171 |
|
172 |
html_output ="<h2>ICD Codes</h2>"
|