Instantaneous1 commited on
Commit
d1407c6
1 Parent(s): d3e79b4
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +1 -1
README.md CHANGED
@@ -13,7 +13,7 @@ pinned: false
13
 
14
  ## Description
15
 
16
- ### A very fast reverse image search webapp in streamlit using NVIDIA's EfficientNet and Facebook's Faiss library over approx 70000 images (224x224). (Trains turbo fast (4.30 miuntes) on 4GB GTX1650 GPU)
17
 
18
  Upload a picture, and AI powered by deep learning will instantly show you visually related matches. Explore and discover connections through the magic of image recognition.
19
 
 
13
 
14
  ## Description
15
 
16
+ ### A very fast reverse image search webapp in streamlit using NVIDIA's EfficientNet and Facebook's Faiss library. 70000 images (224x224), 4GB GTX1650: FAISS GPU in 10 secs vs 30 mins with Annoy CPU
17
 
18
  Upload a picture, and AI powered by deep learning will instantly show you visually related matches. Explore and discover connections through the magic of image recognition.
19
 
app.py CHANGED
@@ -261,7 +261,7 @@ if __name__ == "__main__":
261
  )
262
 
263
  if uploaded_file is not None:
264
- query_image = Image.open(uploaded_file).resize([224, 224])
265
  cropped = st_cropper(query_image, default_coords=[10, 240, 10, 240])
266
  query_image, nearest_neighbors, distances = search_similar_images(
267
  cropped.resize([224, 224]), n_matches
 
261
  )
262
 
263
  if uploaded_file is not None:
264
+ query_image = Image.open(uploaded_file).resize([256, 256])
265
  cropped = st_cropper(query_image, default_coords=[10, 240, 10, 240])
266
  query_image, nearest_neighbors, distances = search_similar_images(
267
  cropped.resize([224, 224]), n_matches