Update app.py

app.py

@@ -166,7 +166,10 @@ def predict_icd(text_input, model_name, label_count):
     label_dict = label_dict_df.set_index('icd9_code')['long_title'].to_dict()
     predicted_labels_float = [float(label) for label in predicted_labels]
     predicted_labels_with_titles = [(label, label_dict.get(str(label), "Not Found")) for label in predicted_labels_float]
-    print("Keys in label_probabilities:", label_probabilities.keys())
+    print("Types of keys in label_probabilities:")
+    for key in label_probabilities.keys():
+        print(type(key))
+    # print("Keys in label_probabilities:", label_probabilities.keys())
 
 
     html_output ="<h2>ICD Codes</h2>"