Update app.py

app.py

@@ -92,7 +92,7 @@ def gradio_find(user_img):
     result_imgs = []
     for i in idxs[:6]:
         img = dataset[int(i)]['image'].resize((128, 128))
-        caption = f"{name} (sim={score:.3f})"
+        caption = f"{name}"
         result_imgs.append((img, caption))
     wiki_text = get_wikipedia_info(name)
     return result_imgs, wiki_text
@@ -100,7 +100,7 @@ with gr.Blocks() as demo:
     gr.Markdown("## Which Celebrity Do You Look Like?")
     with gr.Row():
         img_input = gr.Image(type="pil", label="Upload your face")
-        gallery = gr.Gallery(label="Top 3 similar celebrities", columns=3)
+        gallery = gr.Gallery(label="Similar celebrity", columns=3)
     info_box = gr.Textbox(label="Wikipedia Info", lines=10)
     btn = gr.Button("Find Similar")
     btn.click(fn=gradio_find, inputs=img_input, outputs=[gallery, info_box])