Update app.py

app.py

@@ -96,6 +96,13 @@ with gr.Blocks(css=css) as app:
                             label="Select an object detection model",
                             show_label=True,
                         )
+                    with gr.Row():
+                        options = gr.Dropdown(
+                            value=8,
+                            choices=[2, 4, 8, 16],
+                            label="Select the number of attention heads in encoder and decoder",
+                            show_label=True,
+                        )
                     with gr.Row():
                         slider_input = gr.Slider(
                             minimum=0.2, maximum=1, value=0.7, label="Prediction threshold"