sample-image-search-app

Sample image search app in Japanse.

This repo is used stabilityai/japanese-stable-clip-vit-l-16 and faiss for vector search. The image data is used Flickr 8k Dataset.

Demo

sample-image-search-app-2023-12-10_15.40.00.mp4

Download Models

import os
from huggingface_hub import snapshot_download

hugging_face_token = (
    hugging_face_token if None else os.environ.get("HUGGING_FACE_TOKEN")
)

snapshot_download(repo_id="stabilityai/japanese-stable-clip-vit-l-16", local_dir="models", token=hugging_face_token)

Data pre embedding

$ make embed

Run App

$ docker compose build
$ docker compose up -d

and open localhost:8501