Spaces:

UNIQ-DEV
/

Image-Classification-Benchmark

Sleeping

Image-Classification-Benchmark / src /interface.py

enable multiple image outputs

cebad5c almost 3 years ago

248 Bytes

from .data.classification_result import ClassificationResult
from abc import ABC, abstractmethod
from typing import List

class ModelInterface(ABC):
    @abstractmethod
    def classify_image(self, image) -> List[ClassificationResult]:
        pass

	from .data.classification_result import ClassificationResult
	from abc import ABC, abstractmethod
	from typing import List

	class ModelInterface(ABC):
	@abstractmethod
	def classify_image(self, image) -> List[ClassificationResult]:
	pass