- tweaks

app.py

@@ -11,6 +11,8 @@ model = AutoModel.from_pretrained(model_ckpt)
 hidden_dim = model.config.hidden_size
 dataset = load_dataset("BounharAbdelaziz/Face-Aging-Dataset")
 
+MAX_K = 80
+
 with gr.Blocks() as demo:
     gr.Markdown(
         """
@@ -53,7 +55,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, 30, 10, label="Image Count")
+                adv_image_count = gr.Slider(1, MAX_K, 10, label="Image Count")
                 adv_button = gr.Button("Submit")
 
             with gr.Column(scale=2):
@@ -64,7 +66,7 @@ with gr.Blocks() as demo:
             outputs = model(**inputs)
             embed = outputs.last_hidden_state[0][0]
             result = await index.query(
-                vector=embed.tolist(), top_k=max(1, min(30, count))
+                vector=embed.tolist(), top_k=max(1, min(MAX_K, int(count)))
             )
             return [dataset["train"][int(vector.id)]["image"] for vector in result]
 
@@ -80,4 +82,5 @@ with gr.Blocks() as demo:
         )
 
 if __name__ == "__main__":
-    demo.launch(debug=True, share=True)
+    demo.queue(default_concurrency_limit=40)
+    demo.launch()