File size: 374 Bytes
79859e3
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from __future__ import annotations

from ..models import ModelUtils

class ImageModels():
    def __init__(self, client):
        self.client = client
        self.models = ModelUtils.convert

    def get(self, name, default=None):
        model = self.models.get(name)
        if model and model.best_provider:
            return model.best_provider
        return default