Update app.py

app.py

@@ -268,7 +268,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=1):
             flair_input = gr.File(label="Upload a FLAIR Image (.nii.gz)")
-            use_hd_bet = gr.Checkbox(label="Apply skull-stripping", value=False)
+            hd_bet = gr.Checkbox(label="Apply skull-stripping", value=False)
             submit_button = gr.Button("Submit")
         with gr.Column(scale=2):
             seg_output = gr.File(label="Download the Lesion Segmentation Mask")
@@ -291,7 +291,7 @@ with gr.Blocks() as demo:
 
     submit_button.click(
         fn=run_nnunet_predict,
-        inputs=[flair_input],
+        inputs=[flair_input, hd_bet],
         outputs=[seg_output, input_img, output_img]
     )