From 0f2d6d2ec793c8ae05a31044811d7fc224eb4b38 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Mon, 16 Jan 2023 12:02:37 -0600
Subject: [PATCH] add subscript for fake image

---
 api/onnx_web/upscale.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/api/onnx_web/upscale.py b/api/onnx_web/upscale.py
index 46061c91..c9f11c0d 100644
--- a/api/onnx_web/upscale.py
+++ b/api/onnx_web/upscale.py
@@ -29,6 +29,9 @@ class ONNXImage():
         self.source = source
         self.data = self
 
+    def __getitem__(self, *args):
+        return torch.from_numpy(self.source.__getitem__(*args)).to(torch.float32)
+
     def squeeze(self):
         self.source = np.squeeze(self.source, (0))
         return self