Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -72,7 +72,7 @@ def compute_metrics(pred, gt):
|
|
72 |
return round(mae,4), round(psnr,2), round(ssim,4)
|
73 |
|
74 |
# ---------------- Inference ----------------
|
75 |
-
def infer(image: Image.Image, want_metrics: bool,
|
76 |
if image is None:
|
77 |
return None, None, None, None, None, "", ""
|
78 |
if model is None:
|
|
|
72 |
return round(mae,4), round(psnr,2), round(ssim,4)
|
73 |
|
74 |
# ---------------- Inference ----------------
|
75 |
+
def infer(image: Image.Image, want_metrics: bool, show_L: bool):
|
76 |
if image is None:
|
77 |
return None, None, None, None, None, "", ""
|
78 |
if model is None:
|