52Hz commited on
Commit
3e7c45d
1 Parent(s): 7cb5aa7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -4,12 +4,11 @@ from PIL import Image
4
  # import torch
5
 
6
 
7
- os.system(
8
- 'wget https://github.com/FanChiMao/SUNet/releases/download/0.0/AWGN_denoising_SUNet.pth -P experiments/pretrained_models')
9
- ##
10
 
11
  def inference(img):
12
- os.system('mkdir test')
 
13
  #basewidth = 512
14
  #wpercent = (basewidth / float(img.size[0]))
15
  #hsize = int((float(img.size[1]) * float(wpercent)))
 
4
  # import torch
5
 
6
 
7
+ os.system('wget https://github.com/FanChiMao/SUNet/releases/download/0.0/AWGN_denoising_SUNet.pth -P experiments/pretrained_models')
 
 
8
 
9
  def inference(img):
10
+ # os.system('mkdir test')
11
+ os.makedirs("test", exist_ok=True)
12
  #basewidth = 512
13
  #wpercent = (basewidth / float(img.size[0]))
14
  #hsize = int((float(img.size[1]) * float(wpercent)))