Spaces:

AlexZou
/

Deploy_Restoration

Build error

AlexZou commited on Nov 17, 2022

Commit

ea55f52

•

1 Parent(s): 7970501

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -24,7 +24,7 @@ def inference(image, task):
     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'
 title = "Image Enhancement and Restoration"
 description = ""

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