Image features rebuild.

app.py

@@ -2,6 +2,7 @@ import os
 from io import BytesIO
 import requests
 from datetime import datetime
+import random
 
 # Interface utilities
 import gradio as gr
@@ -52,7 +53,8 @@ def image_from_text(text_input):
     similarities = list((text_features @ photo_features.T).squeeze(0))
     
     ## Return best image :)
-    idx = sorted(zip(similarities, range(photo_features.shape[0])), key=lambda x: x[0], reverse=True)[0][1]
+    idx = sorted(zip(similarities, range(photo_features.shape[0])), key=lambda x: x[0], reverse=True)
+    idx = idx[random.randint(0,4)][1]
     photo_id = photo_ids[idx]
     photo_data = photos[photos["photo_id"] == photo_id].iloc[0]
 

unsplash-dataset/features.npy

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:14ef8a96e6b6adae89432ab046909ab600b5793ba47f2c352168696e7eb9dfb0
-size 51191936
+oid sha256:f375dcc5291457739bccad957f9a418fa7fef87f2c68acc487c58d8bb7672b26
+size 50995328