Update app inputs

app.py

@@ -54,7 +54,7 @@ def create_dl(test_files):
     return loader
 
 # Inference and video generation
-def generate_dicom_video(selected_file):
+def generate_dicom_video(selected_file, n_frames):
     
     test_file = input_files[selected_file]
     test_files = [{'image': test_file}]
@@ -92,21 +92,25 @@ def generate_dicom_video(selected_file):
     return 'dicom.mp4'
 
 
-'''
-test_file = glob.glob('pmrc/SwinUNETR/BTCV/dataset/imagesSampleTs/*.nii.gz')[0]
-generate_dicom_video(test_file)
-'''
+theme = 'dark-peach'
+with gr.Blocks(theme=theme) as demo:
 
-demo = gr.Blocks()
-
-with demo:
+    gr.Markdown('''<center><h1>SwinUnetr BTCV</h1></center>
+	This is a Gradio Blocks app of the winning transformer in the Beyond the Cranial Vault (BTCV) Segmentation Challenge, <a href="https://github.com/darraghdog/Project-MONAI-research-contributions/tree/main/SwinUNETR/BTCV">SwinUnetr (tiny version).</a>.
+	''')
     selected_dicom_key = gr.inputs.Dropdown(
             choices=sorted(input_files),
             type="value",
             label="Select a dicom file")
+    n_frames = gr.Slider(1, 100, value=32, label="Number of dicom slices")
     button_gen_video = gr.Button("Generate Video")
     output_interpolation = gr.Video(label="Generated Video")
-    button_gen_video.click(fn=generate_dicom_video, inputs=selected_dicom_key, outputs=output_interpolation)
+    gr.Markdown(
+            '<center><img src="https://visitor-badge.glitch.me/badge?page_id=gradio-blocks.anime-biggan" alt="visitor badge"/></center>'
+        )
+    button_gen_video.click(fn=generate_dicom_video, 
+                           inputs=[selected_dicom_key, n_frames], 
+                           outputs=output_interpolation)
 
 demo.launch(debug=True, enable_queue=True)