AlexZou commited on
Commit
ce6a6d5
1 Parent(s): 9152760

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -11,6 +11,9 @@ def inference(image, task):
11
  os.system('mkdir tmp')
12
  image.save("tmp/lq_image.png", "PNG")
13
 
 
 
 
14
  if task == 'Dehazing':
15
  os.system("python Dehazing.py --test_path ./tmp/lq_image.png --save_path ./Results/ --pk_path model_zoo/Haze4k.tjm")
16
 
 
11
  os.system('mkdir tmp')
12
  image.save("tmp/lq_image.png", "PNG")
13
 
14
+ if not os.path.exists('Results'):
15
+ os.system('mkdir Results')
16
+
17
  if task == 'Dehazing':
18
  os.system("python Dehazing.py --test_path ./tmp/lq_image.png --save_path ./Results/ --pk_path model_zoo/Haze4k.tjm")
19