download_models.py

import io
import os
import re
import sys

import cv2
import numpy as np
import pyiqa
import timm
import torch
from sentence_transformers import SentenceTransformer

device = "cuda" if torch.cuda.is_available() else "cpu"


image_sim_model = SentenceTransformer("clip-ViT-B-32")