52Hz commited on
Commit
8bf9cb7
·
1 Parent(s): c3d573a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,9 +25,9 @@ def inference(img, model):
25
  elif model == 'Dehazing (dense haze)':
26
  os.system('python main_test_SRMNet.py --input_dir test --task Dehaze_realworld')
27
  elif model == 'Deraining (rainstreak)':
28
- os.system('python main_test_SRMNet.py --input_dir test --task Deraining_raindrop')
29
- elif model == 'Deraining (raindrop)':
30
  os.system('python main_test_SRMNet.py --input_dir test --task Deraining_rainstreak')
 
 
31
  elif model == 'Low-light Enhancement':
32
  os.system('python main_test_SRMNet.py --input_dir test --task LLEnhancement')
33
  elif model == 'Retouching':
 
25
  elif model == 'Dehazing (dense haze)':
26
  os.system('python main_test_SRMNet.py --input_dir test --task Dehaze_realworld')
27
  elif model == 'Deraining (rainstreak)':
 
 
28
  os.system('python main_test_SRMNet.py --input_dir test --task Deraining_rainstreak')
29
+ elif model == 'Deraining (raindrop)':
30
+ os.system('python main_test_SRMNet.py --input_dir test --task Deraining_raindrop')
31
  elif model == 'Low-light Enhancement':
32
  os.system('python main_test_SRMNet.py --input_dir test --task LLEnhancement')
33
  elif model == 'Retouching':