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)]