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}
|