diff --git a/lib/ex_vision/style_transfer/style_transfer.ex b/lib/ex_vision/style_transfer/style_transfer.ex
index b17fa9c..b74f3f0 100644
--- a/lib/ex_vision/style_transfer/style_transfer.ex
+++ b/lib/ex_vision/style_transfer/style_transfer.ex
@@ -23,6 +23,10 @@ for {module, opts} <- Configuration.configuration() do
     #{module} is a custom style transfer model optimised for devices with low computational capabilities and CPU inference.
     """
     require Logger
+
+    @typedoc """
+    A type consisting of output tesnor (stylized image tensor) from style transfer models of shape {#{Enum.join(Tuple.to_list(opts[:resolution]) ++ [3], ", ")}}.
+    """
     @type output_t() :: Nx.Tensor.t()
 
     use ExVision.Model.Definition.Ortex, model: unquote(opts[:model])