Update app.py

app.py

@@ -20,7 +20,7 @@ def get_index(gender, ethnicity, model, no):
 
 def get_nearest_64(gender, ethnicity, model, no):
     ix = get_index(gender, ethnicity, model, no)
-    return ds.select([index[ix][0]])["image"], ds.select(index[ix][1:])["image"]
+    return ds.select([index[ix][0]])["image"], ds.select(index[ix][1:5])["image"]
 
 with gr.Blocks() as demo:
     gender = gr.Radio(gender_labels, label="Gender label")
@@ -29,7 +29,7 @@ with gr.Blocks() as demo:
     no = gr.Radio(nos, label="Image number")
 
     button = gr.Button(value="Get nearest neighbors")
-    button.click(get_nearest_64, inputs=[gender, ethnicity, model, no], outputs=[gr.Image(), gr.Gallery().style(grid=[8])])
+    button.click(get_nearest_64, inputs=[gender, ethnicity, model, no], outputs=[gr.Image(), gr.Gallery()])
 demo.launch()
 # demo = gr.Interface(fn=query_db, inputs="text", outputs=[gr.Image(), gr.Gallery().style(grid=[8])])
 # demo.launch(debug=True)