Skip to content

Commit 02c8907

Browse files
committed
Updated model handler with key
1 parent b024ad5 commit 02c8907

File tree

1 file changed

+38
-4
lines changed

1 file changed

+38
-4
lines changed

RawRefinery/application/ModelHandler.py

Lines changed: 38 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,26 @@
5050
# "affine": True,
5151
# },
5252

53+
"real_blur_kernel_deeper_500": {
54+
"url": None,
55+
"filename": "real_blur_kernel_deeper_500.pt",
56+
"affine": True,
57+
},
58+
"real_blur_kernel_real_noise_100.pt": {
59+
"url": None,
60+
"filename": "real_blur_kernel_real_noise_100.pt",
61+
"affine": True,
62+
},
63+
"real_blur_kernel_real_noise_300.pt": {
64+
"url": None,
65+
"filename": "real_blur_kernel_real_noise_300.pt",
66+
"affine": True,
67+
},
68+
"real_blur_kernel_no_noise_100.pt": {
69+
"url": None,
70+
"filename": "real_blur_kernel_no_noise_100.pt",
71+
"affine": True,
72+
},
5373
"DeepSharpen": {
5474
"url": "https://github.com/rymuelle/RawRefinery/releases/download/v1.2.1-alpha/Deblur_deep_24.pt",
5575
"filename": "Deblur_deep_24.pt",
@@ -196,10 +216,24 @@ def __init__(self):
196216
self.start_time = None
197217
self.model_params = {}
198218

199-
PUBLIC_KEY_PATH = Path(__file__).parent.parent / "verify/model_signing_public_key.pem"
200-
self.pub = serialization.load_pem_public_key(
201-
open(PUBLIC_KEY_PATH, "rb").read()
202-
)
219+
# PUBLIC_KEY_PATH = Path(__file__).parent / "verify/model_signing_public_key.pem"
220+
# self.pub = serialization.load_pem_public_key(
221+
# open(PUBLIC_KEY_PATH, "rb").read()
222+
# )
223+
self.pub = '''-----BEGIN PUBLIC KEY-----
224+
MIICIjANBgkqhkiG9w0BAQEFAAOCAg8AMIICCgKCAgEA8iRGMPqFIFVF0TM/AbMI
225+
DJUqdjY1S7dGn6rYjLixnhohHLKIo2ZhFUfPaeYrDoqJblP9MxbBLm6a782/Us0A
226+
vblTQOsdVFHOlVEiDUkG9CJrzh7arqJF+v2LLP9qPIcL5QdIHM+BCKlbNPBU/TJB
227+
49b6a+1FfKCEeY1z9F8H6GCHGeRB43lz5/1yMoBnq//Rc7NrvinwlNcFYHHM1oj6
228+
Hk6KPkgitya11QgTTva+XimR7cbw7h9/vJKbrS7tValApio3Ypmx7AKf6/k16S9K
229+
BCFDN3cyWmjItQNzEWbO2nuM9d3PX2O4FcZVfsA/GU0qSuKFUrrN0KcxKGglLdu4
230+
3Nt3JmOh+VebVWPSTeMzn2R1LDs2CsDpGG+KnHso80HBBq6RuHTugTiUZ2EwjiXN
231+
lRS7olKFQOPwT0tm1EVkH8IxQgV4KJbCb6hAScvWfsDdsP+bu4R+QI9hfU6HCWG3
232+
a8w1AY+5GT7zp1pzKifmnXgMXF3VnAPTGRhpIvPQfum2+tppLZueXlalobK0MDzi
233+
n36TNhRELao1W7Tvc18fxyZn37BBgKs89JO85/cjD72yhVowW7Hy9lL7RnB+etaN
234+
ehXoYFsJReNmD5KNgRtmXbsCUJ+D8v7BVYNGl1UgebmQnMdMWyiU/3l1Uuy8HS3L
235+
1QJYp42f5QqONttCqVzgzrECAwEAAQ==
236+
-----END PUBLIC KEY-----'''
203237

204238
def load_rh(self, path):
205239
"""Loads the raw file handler"""

0 commit comments

Comments
 (0)