Update app.py

app.py

@@ -24,7 +24,7 @@ def compute_text_embeddings(list_of_strings):
     result = model.get_text_features(**inputs).detach().numpy()
     return result / np.linalg.norm(result, axis=1, keepdims=True)
 
-def image_search(query, corpus, max_results=3):
+def image_search(query, corpus, max_results=24):
     positive_embeddings = None
 
     def concatenate_embeddings(e1, e2):
@@ -68,7 +68,7 @@ def image_search(query, corpus, max_results=3):
         dot_product2 = dot_product2 / np.max(dot_product2, axis=0, keepdims=True)
         dot_product -= np.max(np.maximum(dot_product2, 0), axis=1)
 
-    results = np.argsort(dot_product)[-1 : -n_results - 1 : -1]
+    results = np.argsort(dot_product)[-1 : -max_results - 1 : -1]
     return [
         (
             df[k].iloc[i]["path"],