meghanaraok commited on
Commit
013118e
1 Parent(s): 5255e83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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>"