Update app.py

app.py

@@ -25,7 +25,7 @@ if 'tab' not in st.session_state:
     st.session_state['tab'] = 0
     
 @st.experimental_memo
-def image_search(query, top_k=12):
+def image_search(query, top_k=24):
   with torch.no_grad():
     text_embedding = text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
   _, indices = torch.cosine_similarity(image_embeddings, text_embedding).sort(descending=True)
@@ -54,7 +54,7 @@ def most_similars(embeddings_1, embeddings_2):
   values, indices = torch.cosine_similarity(embeddings_1, embeddings_2).sort(descending=True)
   return values.cpu(), indices.cpu()
 
-def analogy(input_image_path: str,  top_k=12, additional_text: str = '', input_include=True):
+def analogy(input_image_path: str,  top_k=24, additional_text: str = '', input_include=True):
   """ Analogies with embedding space arithmetic.
       Args:
           input_image_path (str): The path to original image 
@@ -68,7 +68,7 @@ def analogy(input_image_path: str,  top_k=12, additional_text: str = '', input_i
 
   return [links[i] for i in indices[:top_k]]
 
-def image_comparison(base_image, top_k=12): 
+def image_comparison(base_image, top_k=24): 
   image_embedding = image_query_embedding(base_image)
   #additional_embedding = text_query_embedding(query=additional_text)
   new_image_embedding = image_embedding #+ additional_embedding