aboutsummaryrefslogtreecommitdiff
path: root/iTexSnip/Utils/TexTellerModel.swift
diff options
context:
space:
mode:
Diffstat (limited to 'iTexSnip/Utils/TexTellerModel.swift')
-rw-r--r--iTexSnip/Utils/TexTellerModel.swift6
1 files changed, 6 insertions, 0 deletions
diff --git a/iTexSnip/Utils/TexTellerModel.swift b/iTexSnip/Utils/TexTellerModel.swift
index fb3bcd9..2f71919 100644
--- a/iTexSnip/Utils/TexTellerModel.swift
+++ b/iTexSnip/Utils/TexTellerModel.swift
@@ -8,6 +8,12 @@
import OnnxRuntimeBindings
import AppKit
+public enum ModelError: Error {
+ case encoderModelNotFound
+ case decoderModelNotFound
+ case imageError
+}
+
public struct TexTellerModel {
public let encoderSession: ORTSession
public let decoderSession: ORTSession