mastari
/

gfpgan-handler

Model card Files Files and versions

mastari commited on Oct 10, 2025

Commit

b51e8ba

·

1 Parent(s): a2eab60

fix_colors

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -124,7 +124,7 @@ class EndpointHandler:
         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")

         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")