Update app.py

app.py

@@ -7,22 +7,22 @@ def inference(image, task):
     image.save("tmp/lq_image.png", "PNG")
     
     if task == 'Dehazing':
-      os.system("python Dehazing.py --test_path ./tmp/lq_image.png --save_path ./results/ --pk_path model_zoo/Haze4k.tjm")
+      os.system("python Dehazing.py --test_path ./tmp/lq_image.png --save_path ./tmp/ --pk_path model_zoo/Haze4k.tjm")
   
     if task == 'LLIE':
-      os.system("python Lowlight.py --test_path ./tmp/lq_image.png --save_path ./results/ --pk_path model_zoo/Low.onnx")
+      os.system("python Lowlight.py --test_path ./tmp/lq_image.png --save_path ./tmp/ --pk_path model_zoo/Low.onnx")
     
     if task == 'SuperResolutionx2':
-      os.system("python SuperResolution.py --test_path ./tmp/lq_image.png --save_path ./results/ --pk_path model_zoo/SRx2.pth --scale 2")
+      os.system("python SuperResolution.py --test_path ./tmp/lq_image.png --save_path ./tmp/ --pk_path model_zoo/SRx2.pth --scale 2")
     
     if task == 'SuperResolutionx3':
-      os.system("python SuperResolution.py --test_path ./tmp/lq_image.png --save_path ./results/ --pk_path model_zoo/SRx3.pth --scale 3")
+      os.system("python SuperResolution.py --test_path ./tmp/lq_image.png --save_path ./tmp/ --pk_path model_zoo/SRx3.pth --scale 3")
       
     if task == 'SuperResolutionx4':
-      os.system("python SuperResolution.py --test_path ./tmp/lq_image.png --save_path ./results/ --pk_path model_zoo/SRx4.pth --scale 4")
+      os.system("python SuperResolution.py --test_path ./tmp/lq_image.png --save_path ./tmp/ --pk_path model_zoo/SRx4.pth --scale 4")
         
-    if task == 'Underwater Enhancement':
-      os.system("python Underwater.py --test_path ./tmp/lq_image.png --save_path ./results/ --pk_path model_zoo/underwater.pth")
+    if task == 'Underwater':
+      os.system("python Underwater.py --test_path ./tmp/lq_image.png --save_path ./tmp/ --pk_path model_zoo/underwater.pth")
       
     return 'results/output.png'