add some transform

app.py

@@ -45,6 +45,7 @@ def image_to_embedding(input_im):
     image = inp.tile(1, 1, 1, 1).to(device=device, dtype=dtype)
     image_embeddings = pipe.image_encoder(image).image_embeds
     image_embeddings = image_embeddings[0]
+    # image_embeddings /= image_embeddings.norm(dim=-1, keepdim=True)
     image_embeddings_np = image_embeddings.cpu().detach().numpy()
     return image_embeddings_np
 
@@ -57,6 +58,7 @@ def prompt_to_embedding(prompt):
     # image = inputs.pixel_values
     with torch.no_grad():
         prompt_embededdings = model.get_text_features(prompt_tokens.to(device))
+    # prompt_embededdings /= prompt_embededdings.norm(dim=-1, keepdim=True)
     prompt_embededdings = prompt_embededdings[0].cpu().detach().numpy()
     return prompt_embededdings
 
@@ -155,7 +157,10 @@ def update_average_embeddings(embedding_base64s_state, embedding_powers):
         # embeddings_b64 = embedding_to_base64(embeddings)
         # return gr.Text.update(embeddings_b64)
         return gr.Text.update('')
+
+    # TODO toggle this to support average or sum
     final_embedding = final_embedding / num_embeddings
+    
     embeddings_b64 = embedding_to_base64(final_embedding)
     return embeddings_b64
 
@@ -255,6 +260,12 @@ tabbed_examples = {
         "540554": "540554.jpeg",
         "Prompt 4": "two trains are coming down the tracks, a steam engine and a modern train.",
     },
+    "Transforms": {
+        "ColorWheel001": "ColorWheel001.jpg",
+        "ColorWheel001 BW": "ColorWheel001 BW.jpg",
+        "ColorWheel002": "ColorWheel002.jpg",
+        "ColorWheel002 BW": "ColorWheel002 BW.jpg",
+    },
     "Portraits": {
         "Snoop": "Snoop Dogg.jpg",
         "Snoop Prompt": "Snoop Dogg",