From b37f8cdd7d4fd876ebe4806f33a17e7871e1dd0c Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Mon, 16 Jan 2023 11:16:06 -0600
Subject: [PATCH] stub out other methods for output image

---
 api/onnx_web/upscale.py | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/api/onnx_web/upscale.py b/api/onnx_web/upscale.py
index 4b0b6430..4c5b97c1 100644
--- a/api/onnx_web/upscale.py
+++ b/api/onnx_web/upscale.py
@@ -25,8 +25,25 @@ resrgan_url = [
 resrgan_path = path.join('..', 'models', 'RealESRGAN_x4plus.onnx')
 
 class ONNXImage():
-    def __init__(self, data) -> None:
-        self.data = data
+    def __init__(self, source) -> None:
+        self.source = source
+        self.data = self
+
+    def squeeze(self):
+        self.source = np.squeeze(self.source, (0))
+        return self
+
+    def float(self):
+        return self
+
+    def cpu(self):
+        return self
+
+    def clamp(self, min, max):
+        self.source = np.clip(self.source, min, max)
+
+    def numpy(self):
+        return self.source
 
 
 class ONNXNet():