CoolFace
Apppublic

ezhdeha/image-deduplication-model

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes
App README

Image Deduplication Model with FAISS

This project detects whether an image is a duplicate of one seen during training on CIFAR-100.

Files

  • resnet18_cifar100.pth: Trained model
  • faiss_index.index: FAISS L2 index
  • threshold.txt: Suggested similarity threshold
  • inference.py: Inference logic
  • app.py: Gradio app for testing

Usage

python
from inference import is_duplicate
with open("image.jpg", "rb") as f:
    result = is_duplicate(f.read())
print(result)