src/models/mobilenet_v3.py

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

class MobilenetV3(ModelInterface):
    
    def __init__(self):
        print('init... mobilenet v3 model')

    def classify_image(self, image) -> List[ClassificationResult]:
        results = [ClassificationResult(class_name=f'example class ({i+1})', confidence=random.uniform(0, 1.0)) for i in range(5)]
        return results