Update app.py

app.py

@@ -171,7 +171,7 @@ with gr.Blocks() as app:
                     label="Misclassified Images", info="Display misclassified images?"
                 )
                 misclassified_count = gr.Dropdown(
-                    choices=["10", "20"],
+                    choices=[str(i + 1) for i in range(20)],
                     label="Select Number of Images",
                     info="Number of Misclassified images",
                     visible=False,
@@ -188,7 +188,7 @@ with gr.Blocks() as app:
                     info="Display GradCAM images?",
                 )
                 gradcam_count = gr.Dropdown(
-                    choices=["10", "20"],
+                    choices=[str(i + 1) for i in range(20)],
                     label="Select Number of Images",
                     info="Number of GradCAM images",
                     interactive=True,