Update app.py

app.py

@@ -24,7 +24,7 @@ with open("urls.txt") as f:
 print("loaded urls")
 assert len(urls) == len(ys)
 d = ys.shape[1]
-index = faiss.IndexFlatL2(d)
+index = faiss.IndexHNSWFlat(d, 32)
 print("building index")
 index.add(ys)
 print('index built')
@@ -54,12 +54,12 @@ row_size = 5
 
 def f(im):
     D, I = index.search(get_emb(im), n_outputs)
-    return [f"Distance: {d}\n![]({urls[i]})" for d, i in zip(D[0], I[0])]
+    return [f"Distance: {d}\n![]({urls[i]}:.1f)" for d, i in zip(D[0], I[0])]
 
 print("preparing gradio")
 with gr.Blocks() as demo:
     gr.Markdown(
-        "# Style Similarity Search\n\nFind artworks with a similar style from a small database (10k artists * 6img/artist)"
+        "# Style Similarity Search\n\nFind artworks with a similar style from a medium database (10k artists * 30 img/artist)"
     )
     img = gr.Image(type="pil", label="Query", height=500)
     btn = gr.Button(variant="primary", value="search")