Update app.py

app.py

@@ -13,7 +13,7 @@ dataset = load_dataset("BounharAbdelaziz/Face-Aging-Dataset")
 
 TOP_K = 1000
 BASE_COUNT=4
-MAX_COUNT = 30
+MAX_COUNT = 10
 
 with gr.Blocks() as demo:
     gr.Markdown(
@@ -57,7 +57,7 @@ with gr.Blocks() as demo:
         with gr.Row():
             with gr.Column(scale=1):
                 adv_input_image = gr.Image(type="pil")
-                adv_image_count = gr.Slider(1, MAX_COUNT, 10, label="Image Count")
+                adv_image_count = gr.Slider(1, MAX_COUNT, BASE_COUNT, label="Image Count")
                 adv_button = gr.Button("Submit")
 
             with gr.Column(scale=2):
@@ -86,9 +86,9 @@ with gr.Blocks() as demo:
         )
         gr.Examples(
             examples=[
-                [dataset["train"][6]["image"], MAX_COUNT],
-                [dataset["train"][7]["image"], MAX_COUNT],
-                [dataset["train"][8]["image"], MAX_COUNT],
+                [dataset["train"][6]["image"], BASE_COUNT],
+                [dataset["train"][7]["image"], BASE_COUNT],
+                [dataset["train"][8]["image"], BASE_COUNT],
             ],
             inputs=[adv_input_image, adv_image_count],
             outputs=adv_output_images,