Update app.py

app.py

@@ -152,7 +152,10 @@ def stage1_process(
     LQ = (LQ[0].permute(1, 2, 0) * 127.5 + 127.5).cpu().numpy().round().clip(0, 255).astype(np.uint8)
     # gamma correction
     LQ = LQ / 255.0
-    LQ = np.power(LQ, gamma_correction)
+    #LQ = np.power(LQ, gamma_correction)
+    LQ = np.power(LQ, float(gamma_correction) if isinstance(gamma_correction, (int, float)) else gamma_correction.value)
+
+
     LQ *= 255.0
     LQ = LQ.round().clip(0, 255).astype(np.uint8)
     print('<<== stage1_process')