Update app.py

app.py

@@ -32,7 +32,7 @@ st.cache(show_spinner=False)
 def image_search(query, top_k=10):
     with torch.no_grad():
         text_embedding = text_encoder(**tokenizer(query, return_tensors='pt')).pooler_output
-    values, indices = (image_embeddings@text_embedding.T).sort(descending=True)
+    values, indices = torch.mm(image_embeddings, text_embedding.T).flatten().sort(descending=True)
     return [links[i] for i in indices[:top_k]]