README.md

metadata

title: StyleSquirrel
emoji: 🐿️
colorFrom: purple
colorTo: blue
sdk: gradio
app_file: app.py
tags:
  - not-for-all-audiences
pinned: false
license: mit

Drop or paste an image → get style tags and see nearest training images.

How it works

• CLIP ViT-B/32 embeds the image
• Trained projector maps to “style space”
• FAISS finds nearest training images
• We tally their style tags and normalize to scores in [0,1]

---

StyleSquirrel

Nearest-neighbor style tagger demo.
Drag an image to see predicted style tags and similar training images.

⚠️ Note on large model files
If you clone this repository locally, you must pull the big model and FAISS index files with Git LFS before running:

git lfs install
git lfs pull