to-be commited on
Commit
edbecfe
1 Parent(s): 9a94b6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
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), Image.ANTIALIAS)
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}