src/models/clip_vit.py

from typing import List
from src.interface import ModelInterface
from src.data.classification_result import ClassificationResult

class ClipVit(ModelInterface):
    def __init__(self):
        print('init... vlip vit model')

    def classify_image(self, image) -> List[ClassificationResult]:
        class_name = "Example Result"
        confidence = 0.85
        return [ClassificationResult(class_name=class_name, confidence=confidence)]