from giskard_vision.core.models.base import ModelBase
import numpy as np

class ModelMyTask(ModelBase):
	def __init__(self, model):
    	    	super().__init__()
    	    	self.model = model

	def predict_rgb_image(self, image: np.ndarray) -> np.ndarray:
    	    	return self.model.predict_rgb_image(image)

	def predict_gray_image(self, image: np.ndarray) -> np.ndarray:
    	    	return self.model.predict_gray_image(image)

mymodel = ...  # Replace with your model
giskard_model = ModelMyTask(model=mymodel)