Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,11 @@ import numpy as np
|
|
12 |
|
13 |
|
14 |
examples = [['sample1.png'],
|
15 |
-
['sample2.png']
|
|
|
|
|
|
|
|
|
16 |
|
17 |
|
18 |
title = "Restormer"
|
@@ -38,7 +42,7 @@ def inference(img):
|
|
38 |
scale = max_res /max(width,height)
|
39 |
width = int(scale*width)
|
40 |
height = int(scale*height)
|
41 |
-
img = img.resize((width,height)
|
42 |
|
43 |
|
44 |
parameters = {'inp_channels':3, 'out_channels':3, 'dim':48, 'num_blocks':[4,6,6,8], 'num_refinement_blocks':4, 'heads':[1,2,4,8], 'ffn_expansion_factor':2.66, 'bias':False, 'LayerNorm_type':'WithBias', 'dual_pixel_task':False}
|
|
|
12 |
|
13 |
|
14 |
examples = [['sample1.png'],
|
15 |
+
['sample2.png'],
|
16 |
+
['sample3.png'],
|
17 |
+
['sample4.png'],
|
18 |
+
['sample5.png'],
|
19 |
+
['sample6.png']]
|
20 |
|
21 |
|
22 |
title = "Restormer"
|
|
|
42 |
scale = max_res /max(width,height)
|
43 |
width = int(scale*width)
|
44 |
height = int(scale*height)
|
45 |
+
img = img.resize((width,height))
|
46 |
|
47 |
|
48 |
parameters = {'inp_channels':3, 'out_channels':3, 'dim':48, 'num_blocks':[4,6,6,8], 'num_refinement_blocks':4, 'heads':[1,2,4,8], 'ffn_expansion_factor':2.66, 'bias':False, 'LayerNorm_type':'WithBias', 'dual_pixel_task':False}
|