Update app.py

app.py

@@ -11,18 +11,23 @@ from torchvision.transforms import (
 from PIL import Image
 import gradio as gr
 
+print("starting...")
 (ys,) = np.load("embs.npz").values()
+print("loaded embs")
 model = torch.load(
     "style-extractor-v0.2.0.ckpt",
     map_location="cpu",
 )
+print("loaded extractor")
 with open("urls.txt") as f:
     urls = f.read().splitlines()
+print("loaded urls")
 assert len(urls) == len(ys)
 d = ys.shape[1]
 index = faiss.IndexFlatL2(d)
-index.is_trained
+print("building index")
 index.add(ys)
+print('index built')
 tf = Compose(
     [
         Resize(
@@ -51,7 +56,7 @@ 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])]
 
-
+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)"
@@ -64,4 +69,5 @@ with gr.Blocks() as demo:
             for _ in range(min(row_size, n_outputs - i * row_size)):
                 outputs.append(gr.Markdown(label=f"#{len(outputs) + 1}"))
     btn.click(f, img, outputs)
+print("starting gradio")
 demo.launch()