Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,11 +9,11 @@ os.system(
|
|
| 9 |
|
| 10 |
def inference(img):
|
| 11 |
os.system('mkdir test')
|
| 12 |
-
basewidth =
|
| 13 |
wpercent = (basewidth / float(img.size[0]))
|
| 14 |
hsize = int((float(img.size[1]) * float(wpercent)))
|
| 15 |
img = img.resize((basewidth, hsize), Image.BILINEAR)
|
| 16 |
-
img.save("test/1.
|
| 17 |
os.system(
|
| 18 |
'python main_test_CMFNet.py --input_dir test --weights experiments/pretrained_models/deraindrop_model.pth')
|
| 19 |
return 'results/1.png'
|
|
|
|
| 9 |
|
| 10 |
def inference(img):
|
| 11 |
os.system('mkdir test')
|
| 12 |
+
basewidth = 512
|
| 13 |
wpercent = (basewidth / float(img.size[0]))
|
| 14 |
hsize = int((float(img.size[1]) * float(wpercent)))
|
| 15 |
img = img.resize((basewidth, hsize), Image.BILINEAR)
|
| 16 |
+
img.save("test/1.png", "PNG")
|
| 17 |
os.system(
|
| 18 |
'python main_test_CMFNet.py --input_dir test --weights experiments/pretrained_models/deraindrop_model.pth')
|
| 19 |
return 'results/1.png'
|