yusufcakmak's picture
feat: test upload - Trendyol DinoV2 Product Similarity and Retrieval Embedding Model
a15fec5 verified
raw
history blame contribute delete
782 Bytes
"""
Trendyol DinoV2 Image Similarity Model
This package contains a fine-tuned DinoV2 model for e-commerce image similarity.
Fully compatible with Hugging Face transformers.
"""
from .modeling_trendyol_dinov2 import TrendyolDinoV2Model, TrendyolDinoV2Config
from .image_processing_trendyol_dinov2 import TrendyolDinoV2ImageProcessor
# Register for AutoModel and AutoImageProcessor
from transformers import AutoConfig, AutoModel, AutoImageProcessor
AutoConfig.register("trendyol_dinov2", TrendyolDinoV2Config)
AutoModel.register(TrendyolDinoV2Config, TrendyolDinoV2Model)
AutoImageProcessor.register(TrendyolDinoV2Config, TrendyolDinoV2ImageProcessor)
__version__ = "1.0.0"
__all__ = [
"TrendyolDinoV2Model",
"TrendyolDinoV2Config",
"TrendyolDinoV2ImageProcessor"
]