Spaces:
Sleeping
Sleeping
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:
|