server/utils/model.py

import os
from dotenv import load_dotenv
from transformers import CLIPProcessor, CLIPModel

load_dotenv()


# Load CLIP model globally for reuse
clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch16")
clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch16")


def get_clip_model():
    return clip_model, clip_processor