mastari
/

gfpgan-handler

Model card Files Files and versions

mastari commited on Oct 10, 2025

Commit

a2eab60

·

1 Parent(s): 4984743

fix_colors

Files changed (1) hide show

handler.py +1 -0

handler.py CHANGED Viewed

@@ -124,6 +124,7 @@ class EndpointHandler:
         logger.info("🖼️ Restoration complete, encoding output...")
         # Encode result as base64
         _, buffer = cv2.imencode(".jpg", restored_img)
         b64_output = base64.b64encode(buffer).decode("utf-8")

         logger.info("🖼️ Restoration complete, encoding output...")
         # Encode result as base64
+	restored_img_rgb = cv2.cvtColor(restored_img, cv2.COLOR_BGR2RGB)
         _, buffer = cv2.imencode(".jpg", restored_img)
         b64_output = base64.b64encode(buffer).decode("utf-8")