Update app.py

app.py

@@ -24,7 +24,7 @@ def inference(image, task):
     if task == 'Underwater Enhancement':
       os.system("python Underwater.py --test_path ./tmp/lq_image.png --save_path ./results/ --pk_path model_zoo/underwater.pth")
       
-    return 'results/lq_image.png'
+    return 'results/output.png'
    
 title = "Image Enhancement and Restoration"
 description = ""