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