Update app.py

app.py

@@ -15,7 +15,7 @@ def inference(img):
     img = img.resize((basewidth, hsize), Image.ANTIALIAS)
     img.save("test/1.png", "PNG")
     os.system(
-        'python main_test_SRMNet.py --weights experiments/pretrained_models/AWGN_denoising_SRMNet.pth')
+        'python main_test_SRMNet.py --input_dir test --weights experiments/pretrained_models/AWGN_denoising_SRMNet.pth')
     return 'result/out.png'