Spaces:
Running
Running
Update app.py
Browse files
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)))
|