Normalize all the things. Also add an example.

app.py

@@ -83,6 +83,7 @@ class Predictor:
             method=model.encode_text,
         )
         emb_from_logits = jax.device_get(emb_from_logits)
+        faiss.normalize_L2(emb_from_logits)
 
         if len(negative_tags_idxs) > 0:
             tags = np.zeros((1, num_classes), dtype=np.float32)
@@ -94,9 +95,10 @@ class Predictor:
                 method=model.encode_text,
             )
             neg_emb_from_logits = jax.device_get(neg_emb_from_logits)
-            emb_from_logits = emb_from_logits - neg_emb_from_logits
+            faiss.normalize_L2(neg_emb_from_logits)
 
-        faiss.normalize_L2(emb_from_logits)
+            emb_from_logits = emb_from_logits - neg_emb_from_logits
+            faiss.normalize_L2(emb_from_logits)
 
         dists, indexes = self.knn_index.search(emb_from_logits, k=n_neighbours)
         neighbours_ids = self.images_ids[indexes][0]
@@ -145,6 +147,28 @@ def main():
 
         similar_images = gr.Gallery(label="Similar images", columns=[5])
 
+        examples = gr.Examples(
+            [
+                [
+                    "artoria_pendragon_(fate),solo",
+                    "excalibur_(fate/stay_night),green_eyes,monochrome,blonde_hair",
+                    ["General", "Sensitive"],
+                    5,
+                    "",
+                    "",
+                ]
+            ],
+            inputs=[
+                positive_tags,
+                negative_tags,
+                selected_ratings,
+                n_neighbours,
+                api_username,
+                api_key,
+            ],
+            outputs=[similar_images],
+        )
+
         find_btn.click(
             fn=predictor.predict,
             inputs=[