Update app.py

app.py

@@ -44,7 +44,7 @@ def find_similar(img, top_k):
         Image.fromarray((X_ref[i].reshape(28, 28) * 255).astype(np.uint8))
         for i in top_idx
     ]
-    return table, best_img, gallery_imgs
+    return table, gallery_imgs
 
 with gr.Blocks() as demo:
 
@@ -69,5 +69,5 @@ with gr.Blocks() as demo:
             interactive=False
         )
 
-    btn.click(find_similar, inputs=[canvas, topk], outputs=[results, best, gallery])
+    btn.click(find_similar, inputs=[canvas, topk], outputs=[results, gallery])
 demo.launch(server_port=7860)