Spaces:

52Hz
/

CMFNet_deraindrop

Sleeping

52Hz commited on Jan 14, 2022

Commit

3aa04f0

1 Parent(s): 5f5224a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ def inference(img):
     img.save("test/1.png", "PNG")
     os.system(
         'python main_test_CMFNet.py --input_dir test --weights experiments/pretrained_models/deraindrop_model.pth')
-    return 'results/out.png'
 title = "Compound Multi-branch Feature Fusion (Deraindrop)"

     img.save("test/1.png", "PNG")
     os.system(
         'python main_test_CMFNet.py --input_dir test --weights experiments/pretrained_models/deraindrop_model.pth')
+    return 'results/1.png'
 title = "Compound Multi-branch Feature Fusion (Deraindrop)"