Update app.py

app.py

@@ -166,9 +166,6 @@ def predict_icd(text_input, model_name, label_count):
     predicted_labels_with_titles = [(label, label_dict.get(str(label), "Not Found")) for label in predicted_labels]
 
     html_output ="<h2>ICD Codes</h2>"
-    # for label, title in predicted_labels_with_titles:
-    #     color = hash_to_color(str(label))
-    #     html_output += f"<div style='background-color: {color}; padding: 5px; margin-bottom: 5px;'>{label} - {title} <span style='float:right;'>{label_probabilities[label]:.2f}%</span></div>"
     for label, title in predicted_labels_with_titles:
         label = str(label)
         color = hash_to_color(label)
@@ -202,8 +199,8 @@ with gr.Blocks() as demo:
     
     model = gr.Radio(["HiLAT", "ClinicalLongformer", "Long-LAT", "Long-HiLAT"], label="Model", info="Select the model used for prediction")
     label_count = gr.Radio(["5", "50"],label = "ICD Code Prediction Range", info="Select the number of top ICD codes to predict: either the top 5 or the top 50")
-    output = gr.HTML(label="ICD Codes")
     predict_btn = gr.Button("Predict")
+    output = gr.HTML(label="ICD Codes")
     predict_btn.click(fn = predict_icd, inputs=[text_input, model, label_count], outputs=[output])
 
 demo.launch(debug= True, share=True)