🌘w🌖

app.py

@@ -32,13 +32,16 @@ def search(query: str, k: int = 4):
         k=k,  # get only top k results
     )
     images = retrieved_examples["image"]
-    labels = {}
-    for i in range(k):
-        labels[retrieved_examples["text"][i]] = scores[i]
+    # labels = {}
+    # for i in range(k):
+    #     labels[retrieved_examples["text"][k-i]] = scores[k-i]
 
-    return images, labels
+    return images #, labels
 
-
-demo = gr.Interface(search, inputs="image", outputs=["gallery", "label"])
+demo = gr.Interface(search, inputs="image", outputs=["gallery"
+                                                     #, "label"
+                                                     ],
+                                                     examples=[("./charmander.jpg",)],
+)
 
 demo.launch(debug=True)