Update app.py

app.py

@@ -313,7 +313,9 @@ def restore_on_gpu(
     with torch.no_grad():
         input_image = upscale_image(input_image, upscale, unit_resolution=32, min_size=min_size)
         LQ = np.array(input_image) / 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)
         LQ = LQ / 255 * 2 - 1